Update personal_info_identifier.py

personal_info_identifier.py

@@ -20,7 +20,7 @@ def highlight_pii(text):
         label = entity['entity_group']
 
         # wrap entity with a span tag with color (red for demo, can be better)
-        span = f'<span style=" padding:2px 4px; border-radius:4px;">{text[entity["start"]:entity["end"]]} <b style="color:red;">[{label}]</b></span>'
+        span = f'<span style="background-color:#ffcccc; padding:2px 4px; border-radius:4px;">{text[entity["start"]:entity["end"]]} <b style="color:red;">[{label}]</b></span>'
         highlighted_text = highlighted_text[:start] + span + highlighted_text[end:]
         offset += len(span) - (end - start)