dslim rayanf commited on
Commit
13e784d
1 Parent(s): fd49b06

Update README.md (#4)

Browse files

- Update README.md (5ee366a9c5b510cdd5bd742efcb7e18cd6485e55)


Co-authored-by: rayan forsat <[email protected]>

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"