Update README.md

README.md

@@ -41,6 +41,7 @@ The model has been first fine-tuned on WikiNER, then focused on the Italian lang
 
 ```python
 from transformers import AutoModelForTokenClassification, AutoTokenizer
+from transformers import pipeline
 import re
 import string
 
@@ -51,26 +52,27 @@ text = "mi chiamo marco rossi, vivo a roma e lavoro per l'agenzia spaziale itali
 
 for p in string.punctuation:
     text = text.replace(p, " " + p + " ")
-    
-ner(text)
+
+ner = pipeline("ner", model=model, tokenizer=tokenizer)
+ner(text, aggregation_strategy="simple")
 
 [{'entity_group': 'PER',
-  'score': 0.9929622,
+  'score': 0.9929623,
   'word': 'marco rossi',
   'start': 9,
   'end': 21},
  {'entity_group': 'LOC',
-  'score': 0.989851,
+  'score': 0.9898509,
   'word': 'roma',
   'start': 31,
   'end': 36},
  {'entity_group': 'ORG',
-  'score': 0.99059105,
+  'score': 0.9905911,
   'word': 'agenzia spaziale italiana',
   'start': 53,
   'end': 79},
  {'entity_group': 'MISC',
-  'score': 0.9247404,
+  'score': 0.92474234,
   'word': 'missione prisma',
   'start': 85,
   'end': 101}]