Commit
·
49657a8
1
Parent(s):
ac41675
Update README.md
Browse files
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("
|
29 |
-
model = AutoModelForTokenClassification.from_pretrained("
|
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)
|