fix chatml

app.py

@@ -33,10 +33,10 @@ def chat(history, system_message, max_tokens, temperature, top_p, top_k, repetit
     history = history or []
 
     messages = BASE_SYSTEM_MESSAGE + system_message.strip() + "\n" + \
-               "\n".join(["\n".join(["User: "+item[0]+"<|end_of_turn|>", "Assistant: "+item[1]+"<|end_of_turn|>"])
+               "\n".join(["\n".join(["<|im_start|>"+"user\n"+item[0]+"<|im_end|>", "<|im_start|>assistant\n"+item[1]+"<|im_end|>"])
                           for item in history])
     # strip the last `<|end_of_turn|>` from the messages
-    messages = messages.rstrip("<|end_of_turn|>")
+    messages = messages.rstrip("<|im_end|>")
     # remove last space from assistant, some models output a ZWSP if you leave a space
     messages = messages.rstrip()