Update README.md
Browse files
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("
|
| 119 |
|
| 120 |
```
|
| 121 |
### Use a pipeline
|
| 122 |
```python
|
| 123 |
from transformers import pipeline
|
| 124 |
|
| 125 |
-
pipe = pipeline("text-classification", model="
|
| 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
|