Update app.py

app.py

@@ -71,12 +71,16 @@ def load_models(inp):
     out_box=gr.update(label=models[inp])
     if "gemma" in models[inp].lower():
         prompt_out="<start_of_turn>userUSER_INPUT<end_of_turn><start_of_turn>model"
+        return out_box,prompt_out, model_state
+
     if "mixtral" in models[inp].lower():
         prompt_out="[INST] USER_INPUT [/INST]"
+        return out_box,prompt_out, model_state
+    
     else:
         prompt_out="USER_INPUT\n"
+        return out_box,prompt_out, model_state
     
-    return out_box,prompt_out, model_state
 
 VERBOSE=False