ajitrajasekharan commited on
Commit
d503dfd
1 Parent(s): 49b6266

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -115,7 +115,7 @@ try:
115
  custom_model_name = st.text_input("Model not listed on left? Type the model name (fill-mask models only)", "")
116
  if (len(custom_model_name) > 0):
117
  model_name = custom_model_name
118
- st.write("Custom model selected:" + model_name)
119
  bert_tokenizer, bert_model = load_bert_model(model_name)
120
  if len(input_text) > 0:
121
  run_test(input_text,top_k)
 
115
  custom_model_name = st.text_input("Model not listed on left? Type the model name (fill-mask models only)", "")
116
  if (len(custom_model_name) > 0):
117
  model_name = custom_model_name
118
+ st.info("Custom model selected:" + model_name)
119
  bert_tokenizer, bert_model = load_bert_model(model_name)
120
  if len(input_text) > 0:
121
  run_test(input_text,top_k)