Update app.py

app.py

@@ -16,19 +16,8 @@ def img2text(url):
 
 # text2story
 def text2story(text):
-    # 优化提示词模板（使用统一的分隔符）
-    prompt_template = f"""[PROMPT_START]
-Write a children's story for ages 3-10 based on: {text}
-Requirements:
-1. Use simple words (1st-3rd grade level)
-2. Main character must be an animal
-3. Include magic elements
-4. Have a happy ending
-5. Story length: 100-120 words
-[PROMPT_END]
-"""
-    
-    # 生成完整提示词（避免重复插入）
+    prompt_template = f"Write a children's story for ages 3-10 based on: {text}"
+        # 生成完整提示词（避免重复插入）
     full_prompt = prompt_template.format(text=text)
     
     # 初始化生成管道
@@ -38,7 +27,7 @@ Requirements:
         max_new_tokens=180,
         min_new_tokens=130,
         temperature=0.7,
-        pad_token_id=50256
+        pad_token_id=50256  # 添加模型专用参数
     )
     
     # 生成原始文本
@@ -64,7 +53,7 @@ Requirements:
 
 # text2audio
 def text2audio(story_text):
-    tts = gTTS(text=story_text, lang='en')
+    tts = gTTS(text = story_text, lang='en')
     audio_bytes = io.BytesIO()
     tts.write_to_fp(audio_bytes)
     audio_bytes.seek(0)