Update app.py

app.py

@@ -7,10 +7,10 @@ def generate_chatml(user_messages, assistant_messages):
     
     for i in range(max_messages):
         if i < len(user_messages) and user_messages[i].strip():
-            chatml += f"<im_start>user\n{user_messages[i].strip()}\n<|im_end>\n"
+            chatml += f"<|im_start|>user\n{user_messages[i].strip()}\n<|im_end|>\n"
         
         if i < len(assistant_messages) and assistant_messages[i].strip():
-            chatml += f"<im_start>assistant\n{assistant_messages[i].strip()}\n<|im_end>\n"
+            chatml += f"<|im_start|>assistant\n{assistant_messages[i].strip()}\n<|im_end|>\n"
     return chatml[15:-11]
 
 with gr.Blocks() as demo: