gabrielandrade2 commited on
Commit
7316338
1 Parent(s): a459477

Replace "cl-tohoku" with "tohoku-nlp"

Browse files
Files changed (2) hide show
  1. config.json +1 -1
  2. predict.py +1 -1
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "cl-tohoku/bert-base-japanese-whole-word-masking",
3
  "architectures": [
4
  "BertForTokenClassification"
5
  ],
1
  {
2
+ "_name_or_path": "tohoku-nlp/bert-base-japanese-whole-word-masking",
3
  "architectures": [
4
  "BertForTokenClassification"
5
  ],
predict.py CHANGED
@@ -148,7 +148,7 @@ def run(model, input, output=None, normalize=False, **kwargs):
148
  bert_tc = classification_model.bert_tc.to(device)
149
 
150
  tokenizer = ner.NER_tokenizer_BIO.from_pretrained(
151
- 'cl-tohoku/bert-base-japanese-whole-word-masking',
152
  num_entity_type=len_num_entity_type # Entityの数を変え忘れないように!
153
  )
154
 
148
  bert_tc = classification_model.bert_tc.to(device)
149
 
150
  tokenizer = ner.NER_tokenizer_BIO.from_pretrained(
151
+ 'tohoku-nlp/bert-base-japanese-whole-word-masking',
152
  num_entity_type=len_num_entity_type # Entityの数を変え忘れないように!
153
  )
154