Update app.py

app.py

@@ -25,8 +25,8 @@ class StopOnTokens(StoppingCriteria):
 
 
 system_role= 'system'
-user_role = 'question'
-assistant_role = "answer"
+user_role = 'user'
+assistant_role = "assistant"
 
 sft_start_token =  "<|im_start|>"
 sft_end_token = "<|im_end|>"
@@ -59,7 +59,7 @@ def predict(message, history):
     
     print(f'model_messages: {model_messages}')
 
-    print(f'model_final_inputs: {tokenizer.apply_chat_template(model_messages, add_generation_prompt=True, tokenize=False)}, flash=True')
+    print(f'model_final_inputs: {tokenizer.apply_chat_template(model_messages, add_generation_prompt=True, tokenize=False)}', flash=True)
     model_inputs = tokenizer.apply_chat_template(model_messages, add_generation_prompt=True, return_tensors="pt").to(device)
     # model_inputs = tokenizer([messages], return_tensors="pt").to(device)
     
@@ -102,7 +102,7 @@ placeholder = """
 </div>
 """
 
-chatbot = gr.Chatbot(label='Sailor', placeholder=None) 
+chatbot = gr.Chatbot(label='OpenCoder', placeholder=None) 
 with gr.Blocks(theme=gr.themes.Soft(), fill_height=True) as demo:
     # gr.Markdown("""<center><font size=8>Sailor-Chat Bot⚓</center>""")
     gr.Markdown("""<p align="center"><img src="https://github.com/sail-sg/sailor-llm/raw/main/misc/wide_sailor_banner.jpg" style="height: 110px"/><p>""")