rayanf commited on
Commit
5ee366a
1 Parent(s): a371ace

Update README.md

Browse files

Fix model name in pipeline from 'base' to 'large'.

Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -57,8 +57,8 @@ You can use this model with Transformers *pipeline* for NER.
57
  from transformers import AutoTokenizer, AutoModelForTokenClassification
58
  from transformers import pipeline
59
 
60
- tokenizer = AutoTokenizer.from_pretrained("dslim/bert-base-NER")
61
- model = AutoModelForTokenClassification.from_pretrained("dslim/bert-base-NER")
62
 
63
  nlp = pipeline("ner", model=model, tokenizer=tokenizer)
64
  example = "My name is Wolfgang and I live in Berlin"
 
57
  from transformers import AutoTokenizer, AutoModelForTokenClassification
58
  from transformers import pipeline
59
 
60
+ tokenizer = AutoTokenizer.from_pretrained("dslim/bert-large-NER")
61
+ model = AutoModelForTokenClassification.from_pretrained("dslim/bert-large-NER")
62
 
63
  nlp = pipeline("ner", model=model, tokenizer=tokenizer)
64
  example = "My name is Wolfgang and I live in Berlin"