Update app.py

app.py

@@ -3,38 +3,11 @@ from transformers import pipeline
 
 pipe = pipeline("text-generation", model="X-D-Lab/MindChat-Qwen2-0_5B")
 
-# 初始化对话历史
-history = []
 
-# 设置 prompt 模板
-def format_prompt(history, user_input):
-    prompt = ""
-    for i, (user, assistant) in enumerate(history):
-        prompt += f"[Round {i+1}]\n问：{user}\n答：{assistant}\n"
-    prompt += f"[Round {len(history)+1}]\n问：{user_input}\n答："
-    return prompt
-
-# 响应函数
 def respond(message):
-    global history
-    prompt = format_prompt(history, message)
-    result = pipe(prompt, max_new_tokens=200, do_sample=True, temperature=0.7)[0]["generated_text"]
-
-    # 提取模型回答部分（去掉提示部分）
-    if "答：" in result:
-        answer = result.split("答：")[-1].strip()
-    else:
-        answer = result.strip()
-
-    # 更新历史
-    history.append((message, answer))
-    return answer
+    result = pipe(message, max_new_tokens=100)
+    return result[0]["generated_text"]
 
-# 重置对话
-def reset():
-    global history
-    history = []
-    return "已重置对话历史。"
 
 # 创建界面
 gr.Interface(