Update run.py

run.py

@@ -87,12 +87,15 @@ client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
 
 def format_prompt(message, history):
-  print("HOSTORY")
+  print("HISTORY")
   print(history)
   prompt = "" #"<s>"
+    c=1
   for user_prompt, bot_response in history:
-    prompt += f"[INST] {user_prompt} [/INST]"
-    prompt += f" {bot_response}</s> "
+    if c<2:
+        prompt += f"[INST] {user_prompt} [/INST]"
+        prompt += f" {bot_response}</s> "
+        c=c+1
   prompt += f"[INST] {message} [/INST]"
   print("Final P")
   print(prompt)