Animal10 Classifier
This model is trained to classify 10 different types of animals using ResNet18.
Model Details:
- Architecture: ResNet18
- Number of classes: 10
- Input size: 224x224 RGB images
How to use:
You can use this model for inference via the Hugging Face Inference API.
from transformers import pipeline
# Load the image classification pipeline
classifier = pipeline("image-classification", model="Mhussnain/animal10-classifier")
# Classify an image
result = classifier("path_to_your_image.jpg")
print(result)
- Downloads last month
- 20
Model tree for Mhussnain/animal10-classifier
Base model
timm/resnet18.a1_in1k