Update app.py

app.py

@@ -25,7 +25,7 @@ def generate_story(caption):
     # 使用 GPU 进行文本生成操作
     story_generator = pipeline(
         "text-generation",
-        model="openai-community/gpt2-medium",
+        model="Qwen/Qwen2-1.5B",
         device=device_id
     )
     messages = (