Update app.py

app.py

@@ -78,15 +78,20 @@ try:
                   label="Original text",
                   value=default_text,
                 )
-      start = time.time()
-      res = get_bert_prediction(input_text,top_k)
+      if st.button("Submit"):
+        with st.spinner("Computing"):
+          start = time.time()
+          try:
+            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 occurred during prediction" + str(e))
+            st.stop()
  
 
 	
@@ -94,7 +99,7 @@ try:
           st.text(f"prediction took {time.time() - start:.2f}s")
 
 except Exception as e:
-  st.error("Some error occured!" + str(e))
+  st.error("Some error occurred during loading" + str(e))
   st.stop()  
 	
 st.write("---")