Update predict.py

predict.py

@@ -62,7 +62,7 @@ def predict_entities(modelpath, sentences_list, len_num_entity_type):
     model = ner.BertForTokenClassification_pl(modelpath, num_labels=81, lr=1e-5) 
     bert_tc = model.bert_tc.to(device)
 
-    MODEL_NAME = 'cl-tohoku/bert-base-japanese-whole-word-masking'
+    MODEL_NAME = 'tohoku-nlp/bert-base-japanese-whole-word-masking'
     tokenizer = ner.NER_tokenizer_BIO.from_pretrained(
         MODEL_NAME,
         num_entity_type =  len_num_entity_type#Entityの数を変え忘れないように！