ajitrajasekharan commited on
Commit
9d7bba2
·
1 Parent(s): 2174a54

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -76,6 +76,7 @@ def run_test(sent,top_k,model_name,display_area):
76
  display_area.text("Loading model:" + st.session_state['model_name'])
77
  st.session_state['bert_tokenizer'], st.session_state['bert_model'] = load_bert_model(st.session_state['model_name'])
78
  try:
 
79
  res = get_bert_prediction(sent,st.session_state['top_k'],st.session_state['model_name'])
80
  return res
81
 
@@ -153,7 +154,7 @@ def main():
153
 
154
  input_status_area = st.empty()
155
  display_area = st.empty()
156
- display_area.text("Currently selected results count = " + str(st.session_state['top_k']) + " " + "Currently selected Model name = " + st.session_state['model_name'])
157
  if submit_button:
158
  start = time.time()
159
  if (len(text_input) == 0):
 
76
  display_area.text("Loading model:" + st.session_state['model_name'])
77
  st.session_state['bert_tokenizer'], st.session_state['bert_model'] = load_bert_model(st.session_state['model_name'])
78
  try:
79
+ display_area.text("Model " + str(st.session_state['model_name']) + " load complete")
80
  res = get_bert_prediction(sent,st.session_state['top_k'],st.session_state['model_name'])
81
  return res
82
 
 
154
 
155
  input_status_area = st.empty()
156
  display_area = st.empty()
157
+ display_area.text("Current selections: results count = " + str(st.session_state['top_k']) + " ; " + " Model name = " + st.session_state['model_name'])
158
  if submit_button:
159
  start = time.time()
160
  if (len(text_input) == 0):