Update app.py

app.py

@@ -158,10 +158,10 @@ def main():
       option = init_selectbox()
       input_text = st.text_input("Enter text below", "",on_change=on_text_change,key='my_text')
       custom_model_name = st.text_input("Model not listed on left? Type the model name (fill-mask BERT models only)", "",key="my_model2",on_change=on_model_change2)
-      #if (len(custom_model_name) > 0):
-       # model_name = custom_model_name
-        #st.info("Custom model selected: " + model_name)
-       # bert_tokenizer, bert_model  = load_bert_model(model_name)
+      if (len(custom_model_name) > 0):
+        model_name = custom_model_name
+        st.info("Custom model selected: " + model_name)
+        bert_tokenizer, bert_model  = load_bert_model(model_name)
       #if len(input_text) > 0:
       #  run_test(input_text,top_k,model_name)
       #else: