span marker example

spanmarker.py

from span_marker import SpanMarkerModel
modelPreTrained = SpanMarkerModel.from_pretrained("tomaarsen/span-marker-mbert-base-multinerd")
modelPreTrained.try_cuda()

def extract_people(text:str):
   entities = modelPreTrained.predict(text)
   full_names = set()
   for entity in entities.ents:
       if entity['label'] == 'PER':
           # Check if the entity has both a first name and a last name
           if len(entity.text.split()) >= 2:
               full_names.add(entity.text)

    return list(full_names)

content = 'text goes here'
entities = extract_people(content)
print(entities)