Update personal_info_identifier.py

personal_info_identifier.py

@@ -11,14 +11,14 @@ pii_detector = pipeline(
 def highlight_pii(text):
     entities = pii_detector(text)
     highlighted_text = text
-    offset = 0  # For shifting positions when inserting tags
+    offset = 0  
 
     for entity in sorted(entities, key=lambda x: x['start']):
         start = entity['start'] + offset
         end = entity['end'] + offset
         label = entity['entity_group']
 
-        # wrap entity with a span tag with color (red for demo, can be better)
+
         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)
@@ -35,7 +35,7 @@ def create_personal_info_tab():
 
         submit_button.click(fn=highlight_pii, inputs=input_text, outputs=output_html)
 
-        # Optional: Add example
+
         gr.Examples(
             examples=[
                 "Hugging Face is a company based in Paris and New York City that acquired Gradio in 2021."