Spaces:

ElvisTsang
/

OnceUponATime

Running

ElvisTsang commited on Mar 4

Commit

37b42cd

verified ·

1 Parent(s): 7d0c379

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

     story_generator = pipeline("text-generation", model="Qwen/Qwen2.5-0.5B", device_map="auto", return_full_text=False)
     prompt = f"Generate a short story under 100 words about {text} without other narrative."
     generated = story_generator(prompt, max_new_tokens=130, do_sample=True, temperature=0.7)
+    story_text = generated[0]['generated_text']
     return story_text
 def text2audio(story_text):