sted97 commited on
Commit
bed6ee7
1 Parent(s): 8021365

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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)