File size: 645 Bytes
5f337a3 cefdac3 d07f52e cefdac3 5f337a3 cefdac3 5f337a3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
import spacy
import gradio as gr
def make_ent_dict(list):
ent_dict = {
'ORG': '',
'PERSON': '',
'GPE': '',
'DATE': '',
'TIME': '',
'MONEY': '',
'PERCENT': '',
'QUANTITY': '',
'PRODUCT': '',
}
for item in list:
if item[0] in ent_dict.keys():
ent_dict[item[0]] = item[1]
return ent_dict
def get_named_entity(text):
nlp = spacy.load("en_core_web_sm")
doc = nlp(text)
ent_list = [(ent.label_, ent.text) for ent in doc.ents]
return make_ent_dict(ent_list)
iface = gr.Interface(fn=get_named_entity, inputs="text", outputs="text")
iface.launch(inline=False)
|