Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSequenceClassification, TextClassificationPipeline
2tok = AutoTokenizer.from_pretrained("celalkartoglu/tr-toxic-bert-multiclass-v1")
3mdl = AutoModelForSequenceClassification.from_pretrained("celalkartoglu/tr-toxic-bert-multiclass-v1")
4pipe = TextClassificationPipeline(model=mdl, tokenizer=tok, top_k=1)
5pipe("Defol git buradan.") # -> [{'label': 'HAKARET', 'score': ...}]
6