dslim
/

bert-large-NER

Token Classification

Model card Files Files and versions

Update README.md

#4

by rayanf - opened Sep 5, 2023

base: refs/heads/main

←

from: refs/pr/4

Discussion Files changed

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -57,8 +57,8 @@ You can use this model with Transformers *pipeline* for NER.
 from transformers import AutoTokenizer, AutoModelForTokenClassification
 from transformers import pipeline
-tokenizer = AutoTokenizer.from_pretrained("dslim/bert-base-NER")
-model = AutoModelForTokenClassification.from_pretrained("dslim/bert-base-NER")
 nlp = pipeline("ner", model=model, tokenizer=tokenizer)
 example = "My name is Wolfgang and I live in Berlin"

 from transformers import AutoTokenizer, AutoModelForTokenClassification
 from transformers import pipeline
+tokenizer = AutoTokenizer.from_pretrained("dslim/bert-large-NER")
+model = AutoModelForTokenClassification.from_pretrained("dslim/bert-large-NER")
 nlp = pipeline("ner", model=model, tokenizer=tokenizer)
 example = "My name is Wolfgang and I live in Berlin"