Update app.py

app.py

@@ -98,7 +98,17 @@ def on_option_change():
   global top_k,model_name
   text = st.session_state.my_choice
   run_test(text,top_k,model_name)
+  
+
+def on_model_change1:
+  model_name = st.session_state.my_model1
+  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
+  st.info("Custom model chosen: " + model_name)
+  bert_tokenizer, bert_model  = load_bert_model(model_name)
   
 def init_selectbox():
   option = st.selectbox(
@@ -124,14 +134,14 @@ print(top_k)
   #  with st.spinner("Computing"):
 try:
             
-      model_name = st.sidebar.selectbox(label='Select Model to Apply',  options=['ajitrajasekharan/biomedical', 'bert-base-cased','bert-large-cased','microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext','allenai/scibert_scivocab_cased','dmis-lab/biobert-v1.1'], index=0,  key = "model_name")
+      model_name = st.sidebar.selectbox(label='Select Model to Apply',  options=['ajitrajasekharan/biomedical', 'bert-base-cased','bert-large-cased','microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext','allenai/scibert_scivocab_cased','dmis-lab/biobert-v1.1'], index=0,  key = "my_model1",on_change=on_model_change1)
       option = init_selectbox()
       input_text = st.text_input("Enter text below", "",on_change=on_text_change,key='my_text')
-      custom_model_name = st.text_input("Model not listed on left? Type the model name (fill-mask BERT models only)", "")
-      if (len(custom_model_name) > 0):
-        model_name = custom_model_name
-        st.info("Custom model selected: " + model_name)
-        bert_tokenizer, bert_model  = load_bert_model(model_name)
+      custom_model_name = st.text_input("Model not listed on left? Type the model name (fill-mask BERT models only)", "",key="my_model2",on_change=on_model_change2)
+      #if (len(custom_model_name) > 0):
+       # model_name = custom_model_name
+        #st.info("Custom model selected: " + model_name)
+       # bert_tokenizer, bert_model  = load_bert_model(model_name)
       #if len(input_text) > 0:
       #  run_test(input_text,top_k,model_name)
       #else: