Added application file

app.py

@@ -0,0 +1,22 @@
+from transformers import AutoTokenizer, AutoModelForTokenClassification
+from transformers.pipelines import pipeline
+import gradio as gr
+import os
+
+
+masker = pipeline(
+    task="token-classification",
+    model="Isotonic/distilbert_finetuned_ai4privacy_v2",
+    device="cpu",
+)
+
+def ai4p_gradio(input):
+    entities = masker(input)
+    return {"text": input, "entities": entities}
+
+demo = gr.Interface(
+    fn=ai4p_gradio, 
+    inputs=gr.Textbox(lines=5, label="Input"), 
+    outputs=gr.HighlightedText(label="output"),
+)
+demo.launch()

requirements.txt

@@ -0,0 +1,4 @@
+torch==2.0.1
+transformers
+gradio
+datasets