CoolFace
Modelpublic

qanastek/51-languages-classifier

sourceHugging Facecc-by-4.0updated 4y agoView on Hugging Face
35likes2.1kdownloads
test.py7 linesDownload Raw Back to root
1from transformers import AutoTokenizer, AutoModelForSequenceClassification, TextClassificationPipeline
2model_name = 'qanastek/51-languages-classifier'
3tokenizer = AutoTokenizer.from_pretrained(model_name)
4model = AutoModelForSequenceClassification.from_pretrained(model_name)
5classifier = TextClassificationPipeline(model=model, tokenizer=tokenizer)
6res = classifier("פרק הבא בפודקאסט בבקשה")
7print(res)