Views
No views yet
1from transformers import AutoTokenizer, AutoModelForTokenClassification
2from transformers import pipeline
3
4tokenizer = AutoTokenizer.from_pretrained("bboninsss/CiceroNER.v1")
5model = AutoModelForTokenClassification.from_pretrained("bboninsss/CiceroNER.v1")
6nlp = pipeline("ner", model=model, tokenizer=tokenizer)
7example = "Mario Rossi è nato a Roma"
8ner_results = nlp(example)
9print(ner_results)