dumitrescustefan commited on
Commit
49657a8
1 Parent(s): ac41675

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -25,8 +25,8 @@ You can use this model with Transformers *pipeline* for NER; you will have to ha
25
  ```python
26
  from transformers import AutoTokenizer, AutoModelForTokenClassification
27
  from transformers import pipeline
28
- tokenizer = AutoTokenizer.from_pretrained("dslim/bert-base-NER")
29
- model = AutoModelForTokenClassification.from_pretrained("dslim/bert-base-NER")
30
  nlp = pipeline("ner", model=model, tokenizer=tokenizer)
31
  example = "Alex cumpără un bilet pentru trenul 3118 în direcția Cluj cu plecare la ora 13:00."
32
  ner_results = nlp(example)
 
25
  ```python
26
  from transformers import AutoTokenizer, AutoModelForTokenClassification
27
  from transformers import pipeline
28
+ tokenizer = AutoTokenizer.from_pretrained("dumitrescustefan/bert-base-romanian-ner")
29
+ model = AutoModelForTokenClassification.from_pretrained("dumitrescustefan/bert-base-romanian-ner")
30
  nlp = pipeline("ner", model=model, tokenizer=tokenizer)
31
  example = "Alex cumpără un bilet pentru trenul 3118 în direcția Cluj cu plecare la ora 13:00."
32
  ner_results = nlp(example)