Update app.py

app.py

@@ -56,9 +56,9 @@ def format_prompt_mixtral(message, history,cust_p):
     return prompt
 
 def format_prompt_choose(message, history, cust_p, model_name):
-    if "gemma" in models[inp].lower():
+    if "gemma" in models[model_name].lower():
         return format_prompt_gemma(message,history,cust_p)
-    if "mixtral" in models[inp].lower():
+    if "mixtral" in models[model_name].lower():
         return format_prompt_mixtral(message,history,cust_p)
     else:
         return format_prompt_default(message,history,cust_p)