Update app.py

app.py

@@ -61,9 +61,9 @@ def format_prompt_mixtral(message, history):
     prompt += f"[INST] {message} [/INST]"
     return prompt
 def format_prompt_choose(message, history, model_name):
-    if "gemma" in model_name.lower():
+    if "gemma" in models[model_name].lower():
         return format_prompt_gemma(message,history)
-    if "mixtral" in model_name.lower():
+    if "mixtral" in models[model_name].lower():
         return format_prompt_mixtral(message,history)
     else:
         return format_prompt_mixtral(message,history)