arubenruben commited on
Commit
d30a731
1 Parent(s): a9ea066

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -26,12 +26,12 @@ import nltk
26
 
27
  ner_classifier = pipeline(
28
  "ner",
29
- model="arubenruben/{REPLACE WITH ONE OF THE PIPELINES}",
30
  device=torch.device("cuda:0") if torch.cuda.is_available() else torch.device("cpu"),
31
  trust_remote_code=True
32
  )
33
 
34
- text = "{INSERT TEXT TO BE CLASSIFIED HERE}"
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
  ```