Update app.py

app.py

@@ -14,7 +14,7 @@ client = Client(
 EOS_STRING = '</s>'
 EOT_STRING = '<EOT>'
 
-def get_prompt(messgae, chat_history, system_prompt):
+def get_prompt(message, chat_history, system_prompt):
     texts = [f"<s>[INST] <<SYS>>\n{system_prompt}\n<</SYS>>\n\n"]
 
     do_strip = False
@@ -103,12 +103,12 @@ with gr.Blocks(theme='Taithrah/Minimal') as demo:
 
     
     with gr.Group():
-        chatbot = gr.Chatbot(label='Mistral-7B-Instruct-v0.1')
+        chatbot = gr.Chatbot(label='Yi-6B')
         with gr.Row():
             textbox = gr.Textbox(
                 container=False,
                 show_label=False,
-                placeholder='Hi, Mistral',
+                placeholder='Hi, Yi',
                 scale=10
             )
             submit_button = gr.Button('Submit', variant='primary', scale=1, min_width=0)