Sterling-Ai commited on
Commit
ec2cbf5
·
verified ·
1 Parent(s): 6386fe6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -115,14 +115,14 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
115
 
116
  tokenizer = AutoTokenizer.from_pretrained("eliasalbouzidi/distilbert-nsfw-text-classifier")
117
 
118
- model = AutoModelForSequenceClassification.from_pretrained("eliasalbouzidi/distilbert-nsfw-text-classifier")
119
 
120
  ```
121
  ### Use a pipeline
122
  ```python
123
  from transformers import pipeline
124
 
125
- pipe = pipeline("text-classification", model="eliasalbouzidi/distilbert-nsfw-text-classifier")
126
  ```
127
 
128
  ## Citation
 
115
 
116
  tokenizer = AutoTokenizer.from_pretrained("eliasalbouzidi/distilbert-nsfw-text-classifier")
117
 
118
+ model = AutoModelForSequenceClassification.from_pretrained("Sterling-Ai/SterlingAI-Video-Prompt-Filter")
119
 
120
  ```
121
  ### Use a pipeline
122
  ```python
123
  from transformers import pipeline
124
 
125
+ pipe = pipeline("text-classification", model="Sterling-Ai/SterlingAI-Video-Prompt-Filter")
126
  ```
127
 
128
  ## Citation