ajitrajasekharan
commited on
Commit
•
49b6266
1
Parent(s):
0feff30
Update app.py
Browse files
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)
|