Views
No views yet
galinette
1from transformers import TokenClassificationPipeline, pipeline
2
3
4MODEL_PATH = "./models/EsperBERTo-small-pos/"
5
6nlp = pipeline(
7 "ner",
8 model=MODEL_PATH,
9 tokenizer=MODEL_PATH,
10)
11# or instantiate a TokenClassificationPipeline directly.
12
13nlp("Mi estas viro kej estas tago varma.")
14
15# {'entity': 'PRON', 'score': 0.9979867339134216, 'word': ' Mi'}
16# {'entity': 'VERB', 'score': 0.9683094620704651, 'word': ' estas'}
17# {'entity': 'VERB', 'score': 0.9797462821006775, 'word': ' estas'}
18# {'entity': 'NOUN', 'score': 0.8509314060211182, 'word': ' tago'}
19# {'entity': 'ADJ', 'score': 0.9996201395988464, 'word': ' varma'}