gliner_zero_shot_named_entity_recognition_quickstart.py

1from gliner import GLiNER
2
3# Initialize GLiNER with the base model
4model = GLiNER.from_pretrained("urchade/gliner_base")
5
6# Sample text for entity extraction
7text = """
8Cristiano Ronaldo dos Santos Aveiro (Portuguese pronunciation: [kɾiʃˈtjɐnu ʁɔˈnaɫdu]; born 5 February 1985) 
9is a Portuguese professional footballer who plays as a forward for Saudi Pro League club Al Nassr 
10and captains the Portugal national team.
11"""
12
13# Define the labels you want to extract
14labels = ["person", "birthday", "location", "organization"]
15
16# Predict entities
17entities = model.predict_entities(text, labels)
18
19# Display results
20for entity in entities:
21    print(f"{entity['text']} => {entity['label']}")