Update app.py

app.py

@@ -82,8 +82,10 @@ def apply_prompt_syntax(prompt, system_prompt, prompt_template, bake_in_prompt_s
         "granite_code - with system": """System:\n{system_prompt}\n\nQuestion:\n{prompt}\n\nAnswer:\n""",
         "granite_code - instruction only": """Question:\n{prompt}\n\nAnswer:\n""",
         ### Mistral Models
-        "mistral_ai models sys": """<s>[INST] System Prompt: {system_prompt} \n\n User Prompt: {prompt} [/INST]\n\n""",
-        "mistral_ai models": """<s>[INST] {prompt} [/INST]\n\n""",
+        "mistral_ai_models_sys": """<s>[INST] System Prompt: {system_prompt} \n\n User Prompt: {prompt} [/INST]\n\n""",
+        "mistral_ai_models": """<s>[INST] {prompt} [/INST]\n\n""",
+        "mistral_ai_small_sys": """<s>[SYSTEM_PROMPT]{system_prompt}[/SYSTEM_PROMPT][INST]{prompt}[/INST]\n\n""",
+        "mistral_ai_small_raw": """[SYSTEM_PROMPT]{system_prompt}[/SYSTEM_PROMPT][INST]{prompt}[/INST]""",
         ### No Syntax
         "no syntax - system": """{system_prompt}\n\n{prompt}""",
         "no syntax - user": """{prompt}""",