Update README.md
Browse files
README.md
CHANGED
@@ -72,7 +72,7 @@ from transformers import pipeline
|
|
72 |
tokenizer = AutoTokenizer.from_pretrained("Babelscape/wikineural-multilingual-ner")
|
73 |
model = AutoModelForTokenClassification.from_pretrained("Babelscape/wikineural-multilingual-ner")
|
74 |
|
75 |
-
nlp = pipeline("ner", model=model, tokenizer=tokenizer)
|
76 |
example = "My name is Wolfgang and I live in Berlin"
|
77 |
|
78 |
ner_results = nlp(example)
|
|
|
72 |
tokenizer = AutoTokenizer.from_pretrained("Babelscape/wikineural-multilingual-ner")
|
73 |
model = AutoModelForTokenClassification.from_pretrained("Babelscape/wikineural-multilingual-ner")
|
74 |
|
75 |
+
nlp = pipeline("ner", model=model, tokenizer=tokenizer, grouped_entities=True)
|
76 |
example = "My name is Wolfgang and I live in Berlin"
|
77 |
|
78 |
ner_results = nlp(example)
|