Update app.py

app.py

@@ -88,10 +88,19 @@ def run_test(sent,top_k,model_name):
             st.stop()
   if start is not None:
     st.text(f"prediction took {time.time() - start:.2f}s")
+    
+def on_text_change():
+  global input_text,top_k,model_name
+  run_test(input_text,top_k,model_name)
+
+def on_option_change():
+  global option,top_k,model_name
+  run_test(option,top_k,model_name)
+
   
 def init_selectbox():
   option = st.selectbox(
-     'Choose any of these sentences or type any text below',
+     'Choose any of these sentences or type any text below',on_change=on_option_change
      ('', "[MASK] who lives in New York and works for XCorp suffers from Parkinson's", "Lou Gehrig who lives in [MASK] and works for XCorp suffers from Parkinson's","Lou Gehrig who lives in New York and works       for [MASK] suffers from Parkinson's","Lou Gehrig who lives in New York and works for XCorp suffers from [MASK]","[MASK] who lives in New York and works for XCorp suffers from Lou Gehrig's", "Parkinson who      lives in [MASK] and works for XCorp suffers from Lou Gehrig's","Parkinson who lives in New York and works for [MASK] suffers from Lou Gehrig's","Parkinson who lives in New York and works for XCorp suffers      from [MASK]","Lou Gehrig","Parkinson","Lou Gehrigh's is a [MASK]","Parkinson is a [MASK]","New York is a [MASK]","New York","XCorp","XCorp is a [MASK]","acute lymphoblastic leukemia","acute lymphoblastic       leukemia is a [MASK]")) 
   return option
 
@@ -107,7 +116,6 @@ top_k = st.sidebar.slider("Select how many predictions do you need", 1 , 50, 20)
 print(top_k)
 
 
-  
 
   #if st.button("Submit"):
     
@@ -116,18 +124,17 @@ 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")
       option = init_selectbox()
-      input_text = st.text_input("Enter text below", "")
+      input_text = st.text_input("Enter text below", "",on_change=on_text_change)
       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)
-      if len(input_text) > 0:
-        run_test(input_text,top_k,model_name)
-        input_text = ""
-      else:
-        if len(option) > 0:
-          run_test(option,top_k,model_name)
+      #if len(input_text) > 0:
+      #  run_test(input_text,top_k,model_name)
+      #else:
+       # if len(option) > 0:
+       #   run_test(option,top_k,model_name)
       if (bert_tokenizer is None):
         bert_tokenizer, bert_model  = load_bert_model(model_name)