Spaces:

adit94
/

entity_extraction

Sleeping

adit94 commited on Jun 14

Commit

3c9c686

•

1 Parent(s): e2a4e07

Update helpers/entity_extraction_helpers.py

Files changed (1) hide show

helpers/entity_extraction_helpers.py CHANGED Viewed

@@ -116,6 +116,7 @@ def process_insurance_document(pii_instance, mongo_instance, openai_instance, oc
     out = {
         "entities":openai_entities,
-        "masked_text":masked_text
     }
     return out

     out = {
         "entities":openai_entities,
+        "masked_text":masked_text,
+        "masked_entities":pii_entities
     }
     return out