Update app.py

app.py

@@ -13,7 +13,14 @@ InferenceClient(models[1]),
 InferenceClient(models[2]),
 InferenceClient(models[3]),
 ]
-
+def format_prompt(message, history):
+    prompt = "<s>"
+    if history:
+        for user_prompt, bot_response in history:
+            prompt += f"[INST] {user_prompt} [/INST]"
+            prompt += f" {bot_response}</s> "
+    prompt += f"[INST] {message} [/INST]"
+    return prompt