Update app.py

app.py

@@ -54,25 +54,22 @@ def get_bert_prediction(input_text,top_k):
   except Exception as error:
     pass
 
-try:
-
-  st.title("Qualitative evaluation of Pretrained BERT models")
-  st.markdown("""
+st.title("Qualitative evaluation of Pretrained BERT models")
+st.markdown("""
         <a href="https://ajitrajasekharan.github.io/2021/01/02/my-first-post.html"><small style="font-size:18px; color: #8f8f8f">This app is used to qualitatively examine the performance of pretrained models to do NER , <b>with no fine tuning</b></small></a>
         """, unsafe_allow_html=True)
-  st.write("Incomplete. Work in progress...")
+st.write("Incomplete. Work in progress...")
   #st.write("https://ajitrajasekharan.github.io/2021/01/02/my-first-post.html")
-  st.write("CLS vectors as well as the model prediction for a blank position are examined")
-
-  top_k = 10
-  print(top_k)
+st.write("CLS vectors as well as the model prediction for a blank position are examined")
+top_k = 10
+print(top_k)
 
   
-  start = None
+start = None
   #if st.button("Submit"):
     
   #  with st.spinner("Computing"):
-   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)
@@ -89,18 +86,16 @@ try:
 
       st.json(res)
 
-    except Exception as e:
-       st.error("Some error occured!" + str(e))
-       st.stop()
 	
-    st.write("---")
-
+ 
 
 	
-    if start is not None:
-        st.text(f"prediction took {time.time() - start:.2f}s")
+      if start is not None:
+          st.text(f"prediction took {time.time() - start:.2f}s")
 
 except Exception as e:
-  print("SOME PROBLEM OCCURED")  
+  st.error("Some error occured!" + str(e))
+  st.stop()  
 	
+   st.write("---")