Update personal_info_identifier.py

personal_info_identifier.py

@@ -1,3 +1,16 @@
 import gradio as gr
+from transformers import pipeline
 
-gr.load("models/iiiorg/piiranha-v1-detect-personal-information")
+# โหลด pipeline
+pii_detector = pipeline("token-classification", model="iiiorg/piiranha-v1-detect-personal-information", aggregation_strategy="simple")
+
+def detect_pii(text):
+    results = pii_detector(text)
+    return results
+
+def create_personal_info_tab():
+    with gr.Column():
+        gr.Markdown("### Detect Personal Information (PII)")
+        textbox = gr.Textbox(label="Enter text", lines=5)
+        output = gr.JSON(label="Detected Entities")
+        textbox.change(fn=detect_pii, inputs=textbox, outputs=output)