# Load model directly
from transformers import AutoImageProcessor, AutoModelForImageClassification
processor = AutoImageProcessor.from_pretrained("BIDEQUITY/autotrain-software_picture_preselection_classifier-2804582686")
model = AutoModelForImageClassification.from_pretrained("BIDEQUITY/autotrain-software_picture_preselection_classifier-2804582686", device_map="auto")Quick Links
Model Trained Using AutoTrain
- Problem type: Multi-class Classification
- Model ID: 2804582686
- CO2 Emissions (in grams): 2.0734
Validation Metrics
- Loss: 0.209
- Accuracy: 0.973
- Macro F1: 0.980
- Micro F1: 0.973
- Weighted F1: 0.973
- Macro Precision: 0.980
- Micro Precision: 0.973
- Weighted Precision: 0.973
- Macro Recall: 0.980
- Micro Recall: 0.973
- Weighted Recall: 0.973
- Downloads last month
- 13
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="BIDEQUITY/autotrain-software_picture_preselection_classifier-2804582686") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")