Commit
·
d30a731
1
Parent(s):
a9ea066
Update README.md
Browse files
README.md
CHANGED
@@ -26,12 +26,12 @@ import nltk
|
|
26 |
|
27 |
ner_classifier = pipeline(
|
28 |
"ner",
|
29 |
-
model="arubenruben/
|
30 |
device=torch.device("cuda:0") if torch.cuda.is_available() else torch.device("cpu"),
|
31 |
trust_remote_code=True
|
32 |
)
|
33 |
|
34 |
-
text = "
|
35 |
tokens = nltk.wordpunct_tokenize(text)
|
36 |
result = ner_classifier(tokens)
|
37 |
```
|
|
|
26 |
|
27 |
ner_classifier = pipeline(
|
28 |
"ner",
|
29 |
+
model="arubenruben/NER-PT-BERT-CRF-Conll2003",
|
30 |
device=torch.device("cuda:0") if torch.cuda.is_available() else torch.device("cpu"),
|
31 |
trust_remote_code=True
|
32 |
)
|
33 |
|
34 |
+
text = "FCPorto vence o Benfica por 5-0 no Estádio do Dragão"
|
35 |
tokens = nltk.wordpunct_tokenize(text)
|
36 |
result = ner_classifier(tokens)
|
37 |
```
|