Update app.py

app.py

@@ -12,15 +12,24 @@ def img2text(url):
 # text2story
 def text2story(text):
     pipe = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2")
-    # 构造更详细的提示，要求生成适合3-10岁儿童阅读的童话故事，并控制输出为50-150 words
+    # 构造 prompt，但只用于内部生成，不让模型在最终输出中重复 prompt
     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_full_text=False，避免将 prompt 原样拼接在输出里
+    # 同时使用 max_new_tokens 控制生成长度，防止生成被截断或过长
+    story_text = pipe(
+        prompt,
+        max_new_tokens=250,       # 根据需要可适当调大，以保证字数
+        min_new_tokens=50,        # 保证至少生成一定长度
+        do_sample=True,
+        top_p=0.95,
+        temperature=0.8,
+        return_full_text=False
+    )[0]['generated_text']
     return story_text