Update app.py

app.py

@@ -75,9 +75,11 @@ def run_test(sent,top_k,model_name,display_area):
   if (st.session_state['bert_tokenizer'] is None):
         display_area.text("Loading model:" + st.session_state['model_name'])
         st.session_state['bert_tokenizer'], st.session_state['bert_model']  = load_bert_model(st.session_state['model_name'])
-  try:
         display_area.text("Model " + str(st.session_state['model_name'])  + " load complete")
+  try:
+        display_area.text("Computing fill-mask prediction...")
         res = get_bert_prediction(sent,st.session_state['top_k'],st.session_state['model_name'])
+        display_area.text("Fill-mask prediction complete")
         return res
             
   except Exception as e: