pt-sk
/

roberta_toxic_classifier

Text Classification

toxic comments classification

Inference Endpoints

Model card Files Files and versions Community

pt-sk commited on Jul 12, 2024

Commit

fdb6cd8

·

verified ·

1 Parent(s): 3ce851e

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -16,8 +16,8 @@ This model is trained for toxicity classification task. The dataset used for tra
 from transformers import RobertaTokenizer, RobertaForSequenceClassification
 # load tokenizer and model weights
-tokenizer = RobertaTokenizer.from_pretrained('SkolkovoInstitute/roberta_toxicity_classifier')
-model = RobertaForSequenceClassification.from_pretrained('SkolkovoInstitute/roberta_toxicity_classifier')
 # prepare the input
 batch = tokenizer.encode('you are amazing', return_tensors='pt')

 from transformers import RobertaTokenizer, RobertaForSequenceClassification
 # load tokenizer and model weights
+tokenizer = RobertaTokenizer.from_pretrained('pt-sk/roberta_toxic_classifier')
+model = RobertaForSequenceClassification.from_pretrained('pt-sk/roberta_toxic_classifier')
 # prepare the input
 batch = tokenizer.encode('you are amazing', return_tensors='pt')