Update app.py

app.py

@@ -26,14 +26,8 @@ def text2story(text):
     )
     
     # 生成并后处理
-    raw_output = pipe(text, return_full_text=False)[0]['generated_text']
-    
-    # 提取核心内容（适配模型输出模式）
-    if "[STORY]:" in raw_output:
-        story = raw_output.split("[STORY]:")[-1].strip()
-    else:
-        story = raw_output
-    
+    story = pipe(text, return_full_text=False)[0]['generated_text']
+
     # 智能截断机制
     target_length = min(max(len(story.split()), 50), 150)
     return " ".join(story.split()[:target_length])