Update app.py

app.py

@@ -6,11 +6,10 @@ client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.3")
 
 # Function to format the conversation history
 def format_prompt(message, history):
-    prompt = "<s>"  # Begin with the start token
+    prompt = "<s>"
     for user_prompt, bot_response in history:
-        # Append each turn of user-bot interaction to the prompt
-        prompt += f"[INST] {user_prompt} [/INST] {bot_response}</s> "
-    prompt += f"[INST] {message} [/INST]"  # Add the latest user message
+        prompt += f"[INST] {user_prompt} [/INST] {bot_response} "
+    prompt += f"[INST] {message} [/INST]</s>"
     return prompt
 
 # Text generation function with parameters