ajitrajasekharan commited on
Commit
314c280
·
1 Parent(s): 1f4b5d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -119,11 +119,12 @@ try:
119
  model_name = custom_model_name
120
  st.info("Custom model selected: " + model_name)
121
  bert_tokenizer, bert_model = load_bert_model(model_name)
122
- if len(input_text) > 0:
123
- run_test(input_text,top_k,model_name)
 
124
  else:
125
- if len(option) > 0:
126
- run_test(option,top_k,model_name)
127
  if (bert_tokenizer is None):
128
  bert_tokenizer, bert_model = load_bert_model(model_name)
129
 
 
119
  model_name = custom_model_name
120
  st.info("Custom model selected: " + model_name)
121
  bert_tokenizer, bert_model = load_bert_model(model_name)
122
+ if len(option) > 0:
123
+ run_test(option,top_k,model_name)
124
+ option = ""
125
  else:
126
+ if len(input_text) > 0:
127
+ run_test(input_text,top_k,model_name)
128
  if (bert_tokenizer is None):
129
  bert_tokenizer, bert_model = load_bert_model(model_name)
130