from transformers import AutoTokenizer, AutoModelForSequenceClassification, TextClassificationPipeline | |
model_name = 'qanastek/51-languages-classifier' | |
tokenizer = AutoTokenizer.from_pretrained(model_name) | |
model = AutoModelForSequenceClassification.from_pretrained(model_name) | |
classifier = TextClassificationPipeline(model=model, tokenizer=tokenizer) | |
res = classifier("פרק הבא בפודקאסט בבקשה") | |
print(res) |