Tirendaz
/

roberta-base-NER

Token Classification

Model card Files Files and versions

Tirendaz commited on Oct 26, 2023

Commit

2c73bd5

·

1 Parent(s): 874c5b1

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -60,8 +60,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("Tirendaz/roberta-base-NER")
+model = AutoModelForTokenClassification.from_pretrained("Tirendaz/roberta-base-NER")
 nlp = pipeline("ner", model=model, tokenizer=tokenizer)
 example = "My name is Wolfgang and I live in Berlin"