Update app.py

app.py

@@ -30,7 +30,12 @@ def write_elo_ratings(elo_ratings):
 def format_chatml_prompt(state):
     chatml_prompt = ""
     for message in state["history"]:
-        chatml_prompt += "|im_start|"+message['role']+": "+ message['content']+"|im_end|\n"
+        j=""
+        if message['role']=='user':
+            j="### Instruction:\n"
+        else:
+            j="### Response:\n"
+        chatml_prompt += j+ message['content']+"\n\n"
     return chatml_prompt+"|im_start|Assistant: "
 def get_bot_response(url, prompt,state):
     chatml_prompt = format_chatml_prompt(state)