Image Classification
Transformers
Safetensors
English
siglip
siglip2
Not-For-All-Audiences
art
Synthetic
nsfw
sfw
Anime Picture
Hentai
Normal
Pornography
Enticing
Sensual
Update README.md
Browse files
README.md
CHANGED
|
@@ -74,7 +74,7 @@ from PIL import Image
|
|
| 74 |
import torch
|
| 75 |
|
| 76 |
# Load model and processor
|
| 77 |
-
model_name = "strangerguardhf/
|
| 78 |
model = SiglipForImageClassification.from_pretrained(model_name)
|
| 79 |
processor = AutoImageProcessor.from_pretrained(model_name)
|
| 80 |
|
|
|
|
| 74 |
import torch
|
| 75 |
|
| 76 |
# Load model and processor
|
| 77 |
+
model_name = "strangerguardhf/nsfw-image-detection" # Replace with your model path if needed
|
| 78 |
model = SiglipForImageClassification.from_pretrained(model_name)
|
| 79 |
processor = AutoImageProcessor.from_pretrained(model_name)
|
| 80 |
|