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