s-nlp
/

roberta_toxicity_classifier

Text Classification

toxic comments classification

Model card Files Files and versions

etomoscow commited on Aug 29, 2024

Commit

121d419

·

verified ·

1 Parent(s): 3cd4508

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('s-nlp/roberta_toxicity_classifier')
+model = RobertaForSequenceClassification.from_pretrained('s-nlp/roberta_toxicity_classifier')
 # prepare the input
 batch = tokenizer.encode('you are amazing', return_tensors='pt')