marquesafonso
commited on
Commit
•
b7c43aa
1
Parent(s):
2800b54
fix example in readme
Browse files
README.md
CHANGED
@@ -25,7 +25,7 @@ model = AutoModelForTokenClassification.from_pretrained("marquesafonso/bertimbau
|
|
25 |
```
|
26 |
from transformers import pipeline
|
27 |
|
28 |
-
pipe = pipeline("
|
29 |
|
30 |
sentence = "Acima de Ederson, abaixo de Rúben Dias. É entre os dois jogadores do Manchester City que se vai colocar Gonçalo Ramos no ranking de vendas mais avultadas do Benfica."
|
31 |
|
@@ -34,7 +34,13 @@ result = pipe([sentence])
|
|
34 |
print(f"{sentence}\n{result}")
|
35 |
|
36 |
# Acima de Ederson, abaixo de Rúben Dias. É entre os dois jogadores do Manchester City que se vai colocar Gonçalo Ramos no ranking de vendas mais avultadas do Benfica.
|
37 |
-
# [[
|
|
|
|
|
|
|
|
|
|
|
|
|
38 |
```
|
39 |
|
40 |
## Acknowledgements
|
|
|
25 |
```
|
26 |
from transformers import pipeline
|
27 |
|
28 |
+
pipe = pipeline("ner", model="marquesafonso/bertimbau-large-ner", aggregation_strategy='simple')
|
29 |
|
30 |
sentence = "Acima de Ederson, abaixo de Rúben Dias. É entre os dois jogadores do Manchester City que se vai colocar Gonçalo Ramos no ranking de vendas mais avultadas do Benfica."
|
31 |
|
|
|
34 |
print(f"{sentence}\n{result}")
|
35 |
|
36 |
# Acima de Ederson, abaixo de Rúben Dias. É entre os dois jogadores do Manchester City que se vai colocar Gonçalo Ramos no ranking de vendas mais avultadas do Benfica.
|
37 |
+
# [[
|
38 |
+
# {'entity_group': 'PESSOA', 'score': 0.99694395, 'word': 'Ederson', 'start': 9, 'end': 16},
|
39 |
+
# {'entity_group': 'PESSOA', 'score': 0.9918462, 'word': 'Rúben Dias', 'start': 28, 'end': 38},
|
40 |
+
# {'entity_group': 'ORGANIZACAO', 'score': 0.96376556, 'word': 'Manchester City', 'start': 69, 'end': 84},
|
41 |
+
# {'entity_group': 'PESSOA', 'score': 0.9993823, 'word': 'Gonçalo Ramos', 'start': 104, 'end': 117},
|
42 |
+
# {'entity_group': 'ORGANIZACAO', 'score': 0.9033079, 'word': 'Benfica', 'start': 157, 'end': 164}
|
43 |
+
# ]]
|
44 |
```
|
45 |
|
46 |
## Acknowledgements
|