Image Classification
Transformers
Safetensors
English
vit
explicit-content-detection
mini
art
sensual-content-detection
Anime
Instructions to use prithivMLmods/vit-mini-explicit-content with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use prithivMLmods/vit-mini-explicit-content with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="prithivMLmods/vit-mini-explicit-content") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")# Load model directly from transformers import AutoImageProcessor, AutoModelForImageClassification processor = AutoImageProcessor.from_pretrained("prithivMLmods/vit-mini-explicit-content") model = AutoModelForImageClassification.from_pretrained("prithivMLmods/vit-mini-explicit-content") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -13,6 +13,7 @@ tags:
|
|
| 13 |
- mini
|
| 14 |
- art
|
| 15 |
- sensual-content-detection
|
|
|
|
| 16 |
---
|
| 17 |
|
| 18 |

|
|
@@ -57,7 +58,7 @@ The model categorizes images into five classes:
|
|
| 57 |
* **Class 3:** Pornography
|
| 58 |
* **Class 4:** Safe for Work
|
| 59 |
|
| 60 |
-
# **Run with Transformers
|
| 61 |
|
| 62 |
```python
|
| 63 |
!pip install -q transformers torch pillow gradio
|
|
|
|
| 13 |
- mini
|
| 14 |
- art
|
| 15 |
- sensual-content-detection
|
| 16 |
+
- Anime
|
| 17 |
---
|
| 18 |
|
| 19 |

|
|
|
|
| 58 |
* **Class 3:** Pornography
|
| 59 |
* **Class 4:** Safe for Work
|
| 60 |
|
| 61 |
+
# **Run with Transformers**
|
| 62 |
|
| 63 |
```python
|
| 64 |
!pip install -q transformers torch pillow gradio
|