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

update get_identified_elements()
jftuga · Jan 3, 2025 · afb0cd1 · afb0cd1
2 parents a778a71 + 020b3de
commit afb0cd1
Showing 1 changed file with 1 addition and 1 deletion.
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: