Merge pull request #6 from jftuga/bugfix-save-tokens

jftuga · web-flow · commit afb0cd1b755a · 2025-01-02T22:17:05.000-05:00
update get_identified_elements()
diff --git a/deidentification/deidentification.py b/deidentification/deidentification.py
@@ -171,7 +171,7 @@ def deidentify_with_wrapped_html(self, text: str, html_begin: str = HTML_BEGIN,
         return buffer.getvalue()
 
     def get_identified_elements(self) -> dict:
-        elements = {"message": self.replaced_text, "entities": self.aggregate_persons, "pronouns": self.aggregate_pronouns}
+        elements = {"message": self.text, "entities": self.aggregate_persons, "pronouns": self.aggregate_pronouns}
         return elements
 
     def _find_all_persons(self) -> int:
