Update app.py

app.py

@@ -12,9 +12,18 @@ def img2text(url):
 # text2story
 def text2story(text):
     pipe = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2")
-    story_text = pipe(text)[0]['generated_text']
+    # 构造更详细的提示，要求生成适合3-10岁儿童阅读的童话故事，并控制输出为50-150 words
+    prompt = (
+        f"Write a complete fairy tale story for children aged 3-10. "
+        f"The story should be engaging, imaginative, and written in clear simple language. "
+        f"Make sure the story is between 50 and 150 words. "
+        f"Use the following scene description as inspiration: {text}"
+    )
+    # 调整生成参数，确保输出长度合适且不被截断
+    story_text = pipe(prompt, max_length=200, min_length=50, do_sample=True, top_p=0.95, temperature=0.8)[0]['generated_text']
     return story_text
 
+
 # text2audio
 def text2audio(story_text):
     pipe = pipeline("text-to-audio", model="Matthijs/mms-tts-eng")