Spaces:

carblacac
/

ner-investing

Sleeping

carblacac commited on Jun 26, 2023

Commit

2c439c5

•

1 Parent(s): 4a8eac5

correct error

Files changed (1) hide show

app.py CHANGED Viewed

@@ -52,7 +52,7 @@ def highlight_text(fileObj):
     entities = nlp(text)
     df = entities_to_df(entities)
-    entities = img_tagging.to_dict('records')
     entities = transform_entity_type(entities)

     entities = nlp(text)
     df = entities_to_df(entities)
+    entities = df.to_dict('records')
     entities = transform_entity_type(entities)