File size: 1,020 Bytes
6514fd7
 
 
 
3cf2fd4
6514fd7
7ddc08e
3cf2fd4
 
6514fd7
2ef326a
775e85f
6514fd7
 
2ef326a
3cf2fd4
6514fd7
3cf2fd4
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
from transformers import pipeline
import gradio as gr

model_checkpoint = "MuntasirHossain/bert-finetuned-ner"
model = pipeline("ner", model=model_checkpoint, aggregation_strategy="simple")

def ner(text):
    output = model(text)
    return {"text": text, "entities": output}

description = "This AI model is trained to identify and classify named entities such as persons (PER), locations (LOC), organizations (ORG) and miscellaneous (MISC) \
in unstructured text." 
title = "Named Entity Recognition"
theme = "grass"
examples=["Mount Everest is Earth's highest mountain, located in the Mahalangur Himal sub-range of the Himalayas. Edmund Hillary and Tenzing Norgay were the \
first climbers confirmed to have reached the summit of Mount Everest on May 29, 1953."]    

gr.Interface(ner,
             gr.Textbox(placeholder="Enter sentence here..."),
             gr.HighlightedText(),
             title=title,
             theme = theme,
             description=description,
             examples=examples).launch()