Update app.py

app.py

@@ -21,11 +21,13 @@ options = {
 # 定义聊天函数
 def respond(message, language, task, max_tokens):
     # 初始化对话历史
-    messages = [{"role": "system", "content": "You are a friendly Chatbot."}]
-    messages.append({"role": "user", "content": message + " " + options[language][task]})
+    system_message = "You are a friendly Chatbot."
+    messages = [{"role": "system", "content": system_message}]
+    user_message = message + " " + options[language][task]
+    messages.append({"role": "user", "content": user_message})
 
     # 编码输入
-    inputs = tokenizer(message + " " + options[language][task], return_tensors="pt", padding=True, truncation=True)
+    inputs = tokenizer(user_message, return_tensors="pt", padding=True, truncation=True)
     # 生成回复
     outputs = model.generate(
         inputs["input_ids"],