diff --git a/README.md b/README.md index 854780a..4d984ce 100644 --- a/README.md +++ b/README.md @@ -44,8 +44,8 @@ Train | Validation ```python from transformers import AutoTokenizer, AutoModelForTokenClassification -tokenizer = AutoTokenizer.from_pretrained("Jean-Baptiste/roberta-large-ner") -model = AutoModelForTokenClassification.from_pretrained("Jean-Baptiste/roberta-large-ner") +tokenizer = AutoTokenizer.from_pretrained("Jean-Baptiste/roberta-large-ner-english") +model = AutoModelForTokenClassification.from_pretrained("Jean-Baptiste/roberta-large-ner-english") ##### Process text sample (from wikipedia)