Update app.py

app.py

@@ -65,17 +65,7 @@ try:
   top_k = 10
   print(top_k)
 
-  model_name = st.sidebar.selectbox(label='Select Model to Apply',  options=['ajitrajasekharan/biomedical', 'BERT'], index=0,  key = "model_name")
-
-  bert_tokenizer, bert_model  = load_bert_model(model_name)
-  default_text = "Imatinib is used to treat"
   
- 
-  input_text = st.text_area(
-    label="Original text",
-    value=default_text,
-  )
-
   start = None
   if st.button("Submit"):
     start = time.time()
@@ -85,6 +75,14 @@ try:
             
 
             try:
+            
+                model_name = st.sidebar.selectbox(label='Select Model to Apply',  options=['ajitrajasekharan/biomedical', 'BERT'], index=0,  key = "model_name")
+                bert_tokenizer, bert_model  = load_bert_model(model_name)
+                default_text = "Imatinib is used to treat"
+                input_text = st.text_area(
+                  label="Original text",
+                  value=default_text,
+                )
                 res = get_bert_prediction(input_text,top_k)