Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 
 def generate_chatml(user_messages, assistant_messages):
     max_messages = max(len(user_messages), len(assistant_messages))
+
+    chatml = ""
     
     for i in range(max_messages):
         if i < len(user_messages) and user_messages[i].strip():
@@ -9,7 +11,7 @@ def generate_chatml(user_messages, assistant_messages):
         
         if i < len(assistant_messages) and assistant_messages[i].strip():
             chatml += f"<im_start>assistant\n{assistant_messages[i].strip()}\n<|im_end>\n"
-    return chatml
+    return chatml[15:-11]
 
 with gr.Blocks() as demo:
     gr.Markdown("# ChatML Template Generator")