Update app.py

app.py

@@ -89,6 +89,11 @@ def run_test(sent,top_k,model_name):
   if start is not None:
     st.text(f"prediction took {time.time() - start:.2f}s")
   
+def init_selectbox():
+  option = st.selectbox(
+     'Choose any of these sentences or type any text below',
+      ('', "[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
 
 st.markdown("<h3 style='text-align: center;'>Qualitative evaluation of  any pretrained BERT model</h3>", unsafe_allow_html=True)
 st.markdown("""
@@ -110,9 +115,7 @@ print(top_k)
 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 = st.selectbox(
-     'Choose any of these sentences or type any text below',
-      ('', "[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]"))  
+      option = init_selectbox()
       input_text = st.text_input("Enter text below", "")
       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):