Update app.py

app.py

@@ -26,10 +26,11 @@ def img2text(url):
         st.error(f"图像描述生成出错: {e}")
         return None
 
-# text2story
+#  text2story 
 def text2story(text):
     try:
-        story_generator = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1", trust_remote_code=True)
+        # 使用 gpt2 模型
+        story_generator = pipeline("text-generation", model="gpt2")
         story = story_generator(text, max_length=200, num_return_sequences=1)[0]['generated_text']
         return story
     except Exception as e: