Update app.py

app.py

@@ -5,6 +5,8 @@ import string
 
 from transformers import BertTokenizer, BertForMaskedLM
 
+st.set_page_config(page_title='Qualitative pretrained model eveluation', page_icon=None, layout='centered', initial_sidebar_state='auto')
+
 @st.cache()
 def load_bert_model(model_name):
   try:
@@ -67,32 +69,29 @@ try:
 
   
   start = None
-  if st.button("Submit"):
-    start = time.time()
-    with st.spinner("Computing"):
+  #if st.button("Submit"):
     
-
-            
-
-            try:
+  #  with st.spinner("Computing"):
+   try:
             
-                model_name = st.sidebar.selectbox(label='Select Model to Apply',  options=['ajitrajasekharan/biomedical', 'BERT'], index=0,  key = "model_name")
+      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)
+      start = time.time()
+      res = get_bert_prediction(input_text,top_k)
 	
 
-                st.header("JSON:")
+      st.header("JSON:")
 
-                st.json(res)
+      st.json(res)
 
-            except Exception as e:
-                st.error("Some error occured!" + str(e))
-                st.stop()
+    except Exception as e:
+       st.error("Some error occured!" + str(e))
+       st.stop()
 	
     st.write("---")