ElvisTsang commited on
Commit
49277f1
·
verified ·
1 Parent(s): 41ca666

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("text-generation", model="Qwen/Qwen2.5-0.5B", device_map="auto", return_full_text=False)
11
- prompt = f"Generate a short story under 100 words about {text}."
12
  generated = story_generator(prompt, max_new_tokens=130, do_sample=True, temperature=0.7)
13
  story_text = generated[0]['generated_text'].replace(prompt, "").strip()
14
  return story_text
 
8
 
9
  def text2story(text):
10
  story_generator = pipeline("text-generation", model="Qwen/Qwen2.5-0.5B", device_map="auto", return_full_text=False)
11
+ prompt = f"Generate a short story under 100 words about {text} without other narrative."
12
  generated = story_generator(prompt, max_new_tokens=130, do_sample=True, temperature=0.7)
13
  story_text = generated[0]['generated_text'].replace(prompt, "").strip()
14
  return story_text