Update app.py

app.py

@@ -114,11 +114,11 @@ def on_model_change1():
   st.info("Pre-selected model chosen: " + model_name)
   bert_tokenizer, bert_model  = load_bert_model(model_name)
 
-def on_model_change2():
-  model_name = st.session_state.my_model2
+def on_model_change2(): 
   global model_name
   global bert_tokenizer
   global bert_model
+  model_name = st.session_state.my_model2
   st.info("Custom model chosen: " + model_name)
   bert_tokenizer, bert_model  = load_bert_model(model_name)