Spaces:

ajitrajasekharan
/

Qualitative-pretrained-model-evaluation

Runtime error

ajitrajasekharan commited on Jan 29, 2022

Commit

8e50f08

1 Parent(s): c6d5fcb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ st.set_page_config(page_title='Qualitative pretrained model eveluation', page_ic
 @st.cache()
 def load_bert_model(model_name):
   try:
-    bert_tokenizer = BertTokenizer.from_pretrained(model_name)
     bert_model = BertForMaskedLM.from_pretrained(model_name).eval()
     return bert_tokenizer,bert_model
   except Exception as e:

 @st.cache()
 def load_bert_model(model_name):
   try:
+    bert_tokenizer = BertTokenizer.from_pretrained(model_name,tolower=False)
     bert_model = BertForMaskedLM.from_pretrained(model_name).eval()
     return bert_tokenizer,bert_model
   except Exception as e: