| import gradio as gr | |
| from transformers import pipeline | |
| ner_pipeline = pipeline("ner", model="dslim/bert-base-NER", grouped_entities=True) | |
| def highlight_entities(text): | |
| entities = ner_pipeline(text) | |
| # Gradio มีระบบ Highlight ข้อความให้อัตโนมัติถ้าเราส่งค่าไปแบบนี้ | |
| return {"text": text, "entities": entities} | |
| demo = gr.Interface( | |
| fn=highlight_entities, | |
| inputs=gr.Textbox(placeholder="Enter sentence here..."), | |
| outputs=gr.HighlightedText(), | |
| title="Name Entity Recognition (NER) Finder" | |
| ) | |
| demo.launch() |