Update app.py

app.py

@@ -74,6 +74,7 @@ def run_test(sent,top_k,model_name):
   start = None
   global bert_tokenizer
   global bert_model
+  global display_container
   if (bert_tokenizer is None):
         bert_tokenizer, bert_model  = load_bert_model(model_name)
   with st.spinner("Computing"):
@@ -82,6 +83,7 @@ def run_test(sent,top_k,model_name):
             res = get_bert_prediction(sent,top_k,model_name)
             st.caption("Results in JSON")
             st.json(res)
+            display_container.write("results")
             
           except Exception as e:
             st.error("Some error occurred during prediction" + str(e))
@@ -139,6 +141,9 @@ try:
        #   run_test(option,top_k,model_name)
       if (bert_tokenizer is None):
         bert_tokenizer, bert_model  = load_bert_model(model_name)
+       
+      display_container = st.container()
+      display_container.write("ajits test")
       
 
 except Exception as e: