qanastek
/

XLMRoberta-Alexa-Intents-NER-NLU

Token Classification Transformers PyTorch

Transformers Token Classification Slot Annotation sequence-tagger-model Inference Endpoints

Model card Files Files and versions Community

qanastek commited on May 9, 2022

Commit

f45f707

•

1 Parent(s): 790fd5d

Update

Files changed (3) hide show

123.png +0 -0
README.md +2 -0
predict.py +2 -1

123.png ADDED Viewed

README.md CHANGED Viewed

@@ -146,6 +146,8 @@ print(res)
 Outputs:
 ```python
 [{'word': '▁neuf', 'score': 0.9911066293716431, 'entity': 'B-time', 'index': 6, 'start': 15, 'end': 19},
 {'word': '▁heures', 'score': 0.9200698733329773, 'entity': 'I-time', 'index': 7, 'start': 20, 'end': 26},

 Outputs:
+![English - Hebrew - Spanish](123.png)
 ```python
 [{'word': '▁neuf', 'score': 0.9911066293716431, 'entity': 'B-time', 'index': 6, 'start': 15, 'end': 19},
 {'word': '▁heures', 'score': 0.9200698733329773, 'entity': 'I-time', 'index': 7, 'start': 20, 'end': 26},

predict.py CHANGED Viewed

@@ -3,7 +3,8 @@ from transformers import AutoTokenizer, AutoModelForTokenClassification, TokenCl
 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("je veux écouter la chanson de jacques brel encore une fois")
 for r in res:
     print(r)

 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)