新增聊天機器人

app.py

@@ -0,0 +1,20 @@
+from transformers import pipeline
+import gradio as gr
+
+generator = pipeline(
+    "text-generation",
+    model="Qwen/Qwen1.5-0.5B-Chat",
+    trust_remote_code=True
+)
+
+chat_history = []
+
+def chat(user_input):
+    chat_history.append(f"使用者: {user_input}")
+    prompt = "\n".join(chat_history[-6:]) + "\nAI:"
+    response = generator(prompt, max_new_tokens=100)[0]["generated_text"]
+    reply = response[len(prompt):].strip()
+    chat_history.append(f"AI: {reply}")
+    return reply
+
+gr.ChatInterface(chat).launch()