Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,8 @@ ner=pipeline("ner",grouped_entities=True)
|
|
| 8 |
|
| 9 |
def generate_entities(input):
|
| 10 |
output= ner(input)
|
| 11 |
-
output
|
|
|
|
| 12 |
return {"text":input,"entities":output}
|
| 13 |
|
| 14 |
|
|
|
|
| 8 |
|
| 9 |
def generate_entities(input):
|
| 10 |
output= ner(input)
|
| 11 |
+
for obj in output:
|
| 12 |
+
obj["entity"]=obj.pop("entity_group")
|
| 13 |
return {"text":input,"entities":output}
|
| 14 |
|
| 15 |
|