Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ def img2text(url):
|
|
8 |
|
9 |
def text2story(text):
|
10 |
story_generator = pipeline("text2text-generation", model="google/flan-t5-xl")
|
11 |
-
prompt = f"Create a short story under 100 words based on: {text}"
|
12 |
generated = story_generator(prompt)
|
13 |
story_text = generated[0]['generated_text']
|
14 |
return story_text
|
|
|
8 |
|
9 |
def text2story(text):
|
10 |
story_generator = pipeline("text2text-generation", model="google/flan-t5-xl")
|
11 |
+
prompt = f"Create a short story under 100 words based on: {text}"
|
12 |
generated = story_generator(prompt)
|
13 |
story_text = generated[0]['generated_text']
|
14 |
return story_text
|