Update app.py

app.py

@@ -6,6 +6,7 @@ import string
 bert_tokenizer = None
 bert_model = None
 global top_k
+model_name = "ajitrajaskharan/biomedical"
 
 
 from transformers import BertTokenizer, BertForMaskedLM
@@ -152,14 +153,13 @@ def main():
     
   #  with st.spinner("Computing"):
   try:
-      
-      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)
+      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)
       init_selectbox()
       st.text_input("Enter text below", "",on_change=on_text_change,key='my_text')
-      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)
+      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)
@@ -169,6 +169,7 @@ def main():
       #if (bert_tokenizer is None):
       #  bert_tokenizer, bert_model  = load_bert_model(model_name)
       st.info("Top k = " + str(top_k))
+      st.info("Model name = " + model_name)