Update app.py

app.py

@@ -67,7 +67,9 @@ def format_prompt_mixtral(message, history):
     prompt += f"[INST] {message} [/INST]"
     return prompt
 
-def format_prompt_choose(message, history, model_name, new_models):
+def format_prompt_choose(message, history, model_name, new_models=None):
+    if not new_models:
+        new_models=models
     if "gemma" in new_models[model_name].lower() and "it" in new_models[model_name].lower():
         return format_prompt_gemma(message,history)
     if "mixtral" in new_models[model_name].lower():