ElvisTsang commited on
Commit
c4e5e01
·
verified ·
1 Parent(s): b2d1149

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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