Update app.py

app.py

@@ -1,21 +1,20 @@
 import gradio as gr
 from transformers import pipeline
 
-# 使用 Qwen-0.5B-Instruct
-generator = pipeline("text-generation", model="Qwen/Qwen2.5-0.5B-Instruct")
+# 使用 DistilGPT2（更小更快）
+generator = pipeline("text-generation", model="distilgpt2")
 
 def chat(message, history):
     result = generator(
         message,
-        max_new_tokens=128,   # 可以稍微長一點，保持流暢
+        max_new_tokens=50,   # 輸出限制更小，加快生成
         do_sample=True,
         temperature=0.7,
         top_p=0.9
     )
-    reply = reply[len(message):].strip()
     reply = result[0]["generated_text"]
     return reply
 
-demo = gr.ChatInterface(fn=chat, title="AI 聊天機器人 (Qwen-0.5B-Instruct)")
+demo = gr.ChatInterface(fn=chat, title="AI 聊天機器人 (DistilGPT2)")
 
 demo.launch()