s-nlp
/

roberta_toxicity_classifier

Text Classification

toxic comments classification

Inference Endpoints

Model card Files Files and versions Community

dardem commited on Oct 14

Commit

71af465

•

1 Parent(s): dc3d961

Update README.md

Files changed (1) hide show

README.md +5 -1

README.md CHANGED Viewed

@@ -6,6 +6,10 @@ tags:
 licenses:
 - cc-by-nc-sa
 license: openrail++
 ---
 ## Toxicity Classification Model
@@ -29,7 +33,7 @@ predicted_label = torch.sigmoid(output.logits).argmax().item()
 ## Citation
-If you use our model in your research, please, cite our work:
 ```
 @inproceedings{logacheva-etal-2022-paradetox,

 licenses:
 - cc-by-nc-sa
 license: openrail++
+base_model:
+- FacebookAI/roberta-large
+datasets:
+- google/jigsaw_toxicity_pred
 ---
 ## Toxicity Classification Model
 ## Citation
+To acknowledge our work, please, use the corresponding citation:
 ```
 @inproceedings{logacheva-etal-2022-paradetox,