rifkat commited on
Commit
7b51b62
1 Parent(s): de7318a
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import pipeline
3
  from transformers import AutoTokenizer, AutoModelForTokenClassification
4
  tokenizer = AutoTokenizer.from_pretrained("51la5/roberta-large-NER")
5
  model = AutoModelForTokenClassification.from_pretrained("51la5/roberta-large-NER")
6
- classifier = pipeline("ner", model=model, tokenizer=tokenizer)
7
 
8
  def get_ner(text):
9
  output = classifier(text)
 
3
  from transformers import AutoTokenizer, AutoModelForTokenClassification
4
  tokenizer = AutoTokenizer.from_pretrained("51la5/roberta-large-NER")
5
  model = AutoModelForTokenClassification.from_pretrained("51la5/roberta-large-NER")
6
+ classifier = pipeline("ner", model=model, tokenizer=tokenizer,grouped_entities=True)
7
 
8
  def get_ner(text):
9
  output = classifier(text)