Create app.py

app.py

@@ -0,0 +1,25 @@
+import gradio as gr
+from transformers import pipeline
+
+# Tải pipeline NER
+ner_model = pipeline("ner", model="mr4/pii-ner-vi")
+
+
+def ner_predict(text):
+    entities = ner_model(text)
+    result = ""
+    for ent in entities:
+        result += f"{ent['word']} ({ent['entity_group']}) - Score: {ent['score']:.2f}\n"
+    return result
+
+# Tạo giao diện
+demo = gr.Interface(
+    fn=ner_predict,
+    inputs=gr.Textbox(lines=5, placeholder="Nhập văn bản tiếng Anh..."),
+    outputs="text",
+    title="Named Entity Recognition Demo",
+    description="Ứng dụng nhận dạng thực thể sử dụng mô hình BERT fine-tuned trên tập dữ liệu CoNLL-2003."
+)
+
+# Chạy ứng dụng
+demo.launch()