Update app.py

app.py

@@ -187,7 +187,7 @@ def format_prompt_mistral(message, history, system_message=system_message,system
 
 def format_prompt_yi(message, history, system_message=system_message,system_understand_message=system_understand_message):
     prompt = (
-        "<s>[INST] <<SYS>>\n" + system_message + "\n<</SYS>>\n\n[/INST]"
+        "<s>[INST] [SYS]\n" + system_message + "\n[/SYS]\n\n[/INST]"
     )
     for user_prompt, bot_response in history:
         prompt += f"[INST] {user_prompt} [/INST]"
@@ -265,7 +265,7 @@ def generate_local(
             sys_system_understand_message = system_understand_message.replace("##LLM_MODEL###",llm_model).replace("##LLM_MODEL_PROVIDER###",llm_provider)
         
         if "yi" in llm_model.lower():
-            formatted_prompt = format_prompt_yi(prompt, history,system_message=sys_message,system_understand_message=sys_system_understand_message)
+            formatted_prompt = format_prompt_mistral(prompt, history,system_message=sys_message,system_understand_message="")
         else:
             formatted_prompt = format_prompt_mistral(prompt, history,system_message=sys_message,system_understand_message=sys_system_understand_message)