Update app.py

app.py

@@ -25,14 +25,14 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
     
     if tokenizer is None or peft_model is None:
         return "Model loading failed. Please check the logs."
-    
+    prompt = ''
     # Construct the prompt
     prompt = system_message
-    for user_msg, assistant_msg in history:
-        if user_msg:
-            prompt += f"\nUser: {user_msg} output: "
-        if assistant_msg:
-            prompt += f"\nAssistant: {assistant_msg}"
+    # for user_msg, assistant_msg in history:
+    #     if user_msg:
+    #         prompt += f"\nUser: {user_msg} output: "
+    #     if assistant_msg:
+    #         prompt += f"\nAssistant: {assistant_msg}"
     prompt += f"\n<user> input:{message} output:"
     
     # Tokenize the input prompt