Update README.md
Browse filesFix model name in pipeline from 'base' to 'large'.
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-
|
61 |
-
model = AutoModelForTokenClassification.from_pretrained("dslim/bert-
|
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"
|