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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -115,6 +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
  bert_tokenizer, bert_model = load_bert_model(model_name)
119
  if len(input_text) > 0:
120
  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.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)