ajitrajasekharan commited on
Commit
07c19db
1 Parent(s): d9f055f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -72,6 +72,7 @@ def get_bert_prediction(input_text,top_k,model_name):
72
  def run_test(sent,top_k,model_name):
73
  start = None
74
  if (st.session_state['bert_tokenizer'] is None):
 
75
  st.session_state['bert_tokenizer'], st.session_state['bert_model'] = load_bert_model(st.session_state['model_name'])
76
  with st.spinner("Computing"):
77
  start = time.time()
@@ -94,6 +95,7 @@ def on_text_change():
94
  def on_option_change():
95
 
96
  text = st.session_state.my_choice
 
97
  run_test(text,st.session_state['top_k'],st.session_state['model_name'])
98
 
99
  def on_results_count_change():
 
72
  def run_test(sent,top_k,model_name):
73
  start = None
74
  if (st.session_state['bert_tokenizer'] is None):
75
+ st.info("Loading model:",st.session_state['model_name'])
76
  st.session_state['bert_tokenizer'], st.session_state['bert_model'] = load_bert_model(st.session_state['model_name'])
77
  with st.spinner("Computing"):
78
  start = time.time()
 
95
  def on_option_change():
96
 
97
  text = st.session_state.my_choice
98
+ st.info("Preselected text chosen")
99
  run_test(text,st.session_state['top_k'],st.session_state['model_name'])
100
 
101
  def on_results_count_change():