Instructions to use AdamCodd/vit-base-nsfw-detector with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers.js
How to use AdamCodd/vit-base-nsfw-detector with Transformers.js:
// npm i @huggingface/transformers import { pipeline } from '@huggingface/transformers'; // Allocate pipeline const pipe = await pipeline('image-classification', 'AdamCodd/vit-base-nsfw-detector'); - Transformers
How to use AdamCodd/vit-base-nsfw-detector with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="AdamCodd/vit-base-nsfw-detector") 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("AdamCodd/vit-base-nsfw-detector") model = AutoModelForImageClassification.from_pretrained("AdamCodd/vit-base-nsfw-detector") - Notebooks
- Google Colab
- Kaggle
What's included in the NSFW classification ?
#4
by LDanielBlueway - opened
Hello
I want to use this model to filter helpdesk support request. From what I see on the model card, it includes sexy/porn related content. But does it also include gore, extremist symbol (Nazis cross, ect) and grafic violence?
Hello,
Yes, the model was trained on sexual content. For gore, extremism, violence you'll need to use another model trained on that content.
Ok, thanks