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) |