Spaces:

chanukool
/

ner-tagger

Runtime error

chanukool commited on Aug 6, 2022

Commit

582db31

1 Parent(s): 0b556bc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,10 +4,11 @@ from transformers import pipeline
 title="Named Entity Recognition"
 description="NER"
-ner=pipeline("ner")
 def generate_entities(input):
   output= ner(input)
   return {"text":input,"entities":output}

 title="Named Entity Recognition"
 description="NER"
+ner=pipeline("ner",grouped_entities=True)
 def generate_entities(input):
   output= ner(input)
+  output['entity']=output.pop('entity_group')
   return {"text":input,"entities":output}