Update app.py

app.py

@@ -91,7 +91,7 @@ def load_model(_docs):
     #     return f"""[INST] <<SYS>>{system_prompt}<</SYS>>{prompt} [/INST]""".strip()
 
     def generate_prompt(prompt: str, system_prompt: str = DEFAULT_SYSTEM_PROMPT) -> str:
-        return f"""<|im_start|>system{system_prompt}<|im_end|><|im_start|>user{prompt}<|im_end|>""".strip()
+        return f"""<s>[INST] <<SYS>>{{ system_prompt }}<</SYS>>{{ prompt }} [/INST]""".strip()
 
 
     streamer = TextStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)