update

app.py

@@ -26,11 +26,11 @@ def generate_text(
     top_p,
 ):
     temp = ""
-    input_prompt = f"\n{system_message}\n "
+    input_prompt = f"[INST] <<SYS>>\n{system_message}\n<</SYS>>\n\n "
     for interaction in history:
-        input_prompt = input_prompt + str(interaction[0]) + str(interaction[1])
+        input_prompt = input_prompt + str(interaction[0]) + " [/INST] " + str(interaction[1]) + " </s><s> [INST] "
 
-    input_prompt = input_prompt + str(message)
+    input_prompt = input_prompt + str(message) + " [/INST] "
 
     output = llm(
         input_prompt,