File size: 544 Bytes
b0a8168
 
 
 
 
f45f707
 
b0a8168
 
 
1
2
3
4
5
6
7
8
9
10
from transformers import AutoTokenizer, AutoModelForTokenClassification, TokenClassificationPipeline

tokenizer = AutoTokenizer.from_pretrained('qanastek/XLMRoberta-Alexa-Intents-NER-NLU')
model = AutoModelForTokenClassification.from_pretrained('qanastek/XLMRoberta-Alexa-Intents-NER-NLU')
predict = TokenClassificationPipeline(model=model, tokenizer=tokenizer)
res = predict("réveille-moi à neuf heures du matin le vendredi")
# res = predict("je veux écouter la chanson de jacques brel encore une fois")

for r in res:
    print(r)