Instructions to use jaranohaal/vit-base-violence-detection with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use jaranohaal/vit-base-violence-detection with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="jaranohaal/vit-base-violence-detection") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")# Load model directly from transformers import AutoModelForImageClassification model = AutoModelForImageClassification.from_pretrained("jaranohaal/vit-base-violence-detection", device_map="auto") - Inference
- Notebooks
- Google Colab
- Kaggle
Having a dependency error while running the model
#7
by akhiltenny - opened
i used the exact same code in the article and this happened:
ImportError: cannot import name 'ViTFeatureExtractor' from 'transformers'
i later asked chat gpt and got i should use "AutoImageProcessor" instead of "ViTFeatureExtractor", and i used and i got a big error looks like in the image