Downloads · 30 days
9
43% of all-time downloads
Hemanthhebbal/custom-image-classifier
custom-image-classifier is a image classification model from Hemanthhebbal. Use it when you need a label for an image. The card lists the license as apache-2.0.
This is a PyTorch Computer Vision Image Classification Model fine-tuned for high-performance inference.
Downloads · 30 days
9
43% of all-time downloads
All-time downloads
21
Public
Parameters
2.3M
9.1 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors9.1 MB · 100%
From the Hugging Face model README
This is a PyTorch Computer Vision Image Classification Model fine-tuned for high-performance inference.
from transformers import pipeline
classifier = pipeline("image-classification", model="Hemanthhebbal/custom-image-classifier")
result = classifier("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/beets.jpg")
print(result)