mlibbot-backend / utils /spacy_ner.py
RafaelYefta's picture
Upload 49 files
c237f1b verified
raw
history blame contribute delete
183 Bytes
import spacy
nlp = spacy.load("xx_ent_wiki_sm")
def extract_entities(text: str):
doc = nlp(text)
ents = [(ent.text, ent.label_) for ent in doc.ents]
return ents