zjunlp
/

SafeEdit-Safety-Classifier

Text Classification

Inference Endpoints

Model card Files Files and versions Community

mengru commited on May 28

Commit

e1fe635

•

1 Parent(s): b128382

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -10,7 +10,7 @@ license: apache-2.0
 ```shell
 from transformers import RobertaForSequenceClassification, RobertaTokenizer
-safety_classifier_dir = 'zjunlp/DINM-Safety-Classifier'
 safety_classifier_model = RobertaForSequenceClassification.from_pretrained(safety_classifier_dir)
 safety_classifier_tokenizer = RobertaTokenizer.from_pretrained(safety_classifier_dir)
 ```

 ```shell
 from transformers import RobertaForSequenceClassification, RobertaTokenizer
+safety_classifier_dir = 'zjunlp/SafeEdit-Safety-Classifier'
 safety_classifier_model = RobertaForSequenceClassification.from_pretrained(safety_classifier_dir)
 safety_classifier_tokenizer = RobertaTokenizer.from_pretrained(safety_classifier_dir)
 ```