s-nlp
/

roberta_toxicity_classifier

Text Classification

toxic comments classification

Model card Files Files and versions

dardem commited on Nov 8, 2024

Commit

048c25b

·

verified ·

1 Parent(s): 71af465

Update README.md

Files changed (1) hide show

README.md +1 -2

README.md CHANGED Viewed

@@ -27,8 +27,7 @@ model = RobertaForSequenceClassification.from_pretrained('s-nlp/roberta_toxicity
 batch = tokenizer.encode("You are amazing!", return_tensors="pt")
 output = model(batch)
-predicted_label = torch.sigmoid(output.logits).argmax().item()
-# 0 for neutral, 1 for toxic
 ```
 ## Citation

 batch = tokenizer.encode("You are amazing!", return_tensors="pt")
 output = model(batch)
+# idx 0 for neutral, idx 1 for toxic
 ```
 ## Citation