Hugging Face's logo

zjunlp
/
SafeEdit-Safety-Classifier

Text Classification
Transformers
PyTorch
roberta
Inference Endpoints
Model card Files Community
1
mengru commited on
Commit
e1fe635
·
verified ·
1 Parent(s): b128382

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -10,7 +10,7 @@ license: apache-2.0
10
 
11
  ```shell
12
  from transformers import RobertaForSequenceClassification, RobertaTokenizer
13
- safety_classifier_dir = 'zjunlp/DINM-Safety-Classifier'
14
  safety_classifier_model = RobertaForSequenceClassification.from_pretrained(safety_classifier_dir)
15
  safety_classifier_tokenizer = RobertaTokenizer.from_pretrained(safety_classifier_dir)
16
  ```
 
10
 
11
  ```shell
12
  from transformers import RobertaForSequenceClassification, RobertaTokenizer
13
+ safety_classifier_dir = 'zjunlp/SafeEdit-Safety-Classifier'
14
  safety_classifier_model = RobertaForSequenceClassification.from_pretrained(safety_classifier_dir)
15
  safety_classifier_tokenizer = RobertaTokenizer.from_pretrained(safety_classifier_dir)
16
  ```