Update app.py

app.py

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