marquesafonso commited on
Commit
b7c43aa
1 Parent(s): 2800b54

fix example in readme

Browse files
Files changed (1) hide show
  1. README.md +8 -2
README.md CHANGED
@@ -25,7 +25,7 @@ model = AutoModelForTokenClassification.from_pretrained("marquesafonso/bertimbau
25
  ```
26
  from transformers import pipeline
27
 
28
- pipe = pipeline("token-classification", model="marquesafonso/bertimbau-large-ner")
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
- # [[{'entity': 'B-PESSOA', 'score': 0.99976975, 'index': 4, 'word': 'Ed', 'start': 9, 'end': 11}, {'entity': 'I-PESSOA', 'score': 0.9941182, 'index': 5, 'word': '##erson', 'start': 11, 'end': 16}, {'entity': 'B-PESSOA', 'score': 0.9998306, 'index': 9, 'word': 'R', 'start': 28, 'end': 29}, {'entity': 'I-PESSOA', 'score': 0.9737293, 'index': 10, 'word': '##ú', 'start': 29, 'end': 30}, {'entity': 'I-PESSOA', 'score': 0.9944133, 'index': 11, 'word': '##ben', 'start': 30, 'end': 33}, {'entity': 'I-PESSOA', 'score': 0.9994117, 'index': 12, 'word': 'Dias', 'start': 34, 'end': 38}, {'entity': 'B-ORGANIZACAO', 'score': 0.94043595, 'index': 20, 'word': 'Manchester', 'start': 69, 'end': 79}, {'entity': 'I-ORGANIZACAO', 'score': 0.9870952, 'index': 21, 'word': 'City', 'start': 80, 'end': 84}, {'entity': 'B-PESSOA', 'score': 0.9997652, 'index': 26, 'word': 'Gonçalo', 'start': 104, 'end': 111}, {'entity': 'I-PESSOA', 'score': 0.9989994, 'index': 27, 'word': 'Ramos', 'start': 112, 'end': 117}, {'entity': 'B-ORGANIZACAO', 'score': 0.9033079, 'index': 37, 'word': 'Benfica', 'start': 157, 'end': 164}]]
 
 
 
 
 
 
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