Image Classification with AutoTrain
NimaBoscarino
• • 2# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("NimaBoscarino/butterflies", device_map="auto")Quick Links
Configuration Parsing Warning:Config file config.json cannot be fetched (too big)
Model Trained Using AutoTrain
- Problem type: Multi-class Classification
- Model ID: 1433953639
- CO2 Emissions (in grams): 0.0117
Validation Metrics
- Loss: 0.606
- Accuracy: 0.880
- Macro F1: 0.844
- Micro F1: 0.880
- Weighted F1: 0.844
- Macro Precision: 0.827
- Micro Precision: 0.880
- Weighted Precision: 0.827
- Macro Recall: 0.880
- Micro Recall: 0.880
- Weighted Recall: 0.880
- Downloads last month
- 10
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="NimaBoscarino/butterflies") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")