# Load model directly
from transformers import AutoImageProcessor, AutoModelForImageClassification
processor = AutoImageProcessor.from_pretrained("bortle/astrophotography-object-classifier-alpha2")
model = AutoModelForImageClassification.from_pretrained("bortle/astrophotography-object-classifier-alpha2")Quick Links
Model Trained Using AutoTrain
- Problem type: Multi-class Classification
- Model ID: 40327104863
- CO2 Emissions (in grams): 0.0118
Validation Metrics
- Loss: 0.013
- Accuracy: 1.000
- Macro F1: 1.000
- Micro F1: 1.000
- Weighted F1: 1.000
- Macro Precision: 1.000
- Micro Precision: 1.000
- Weighted Precision: 1.000
- Macro Recall: 1.000
- Micro Recall: 1.000
- Weighted Recall: 1.000
- Downloads last month
- 1
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="bortle/astrophotography-object-classifier-alpha2") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")