# Load model directly
from transformers import AutoImageProcessor, AutoModelForImageClassification
processor = AutoImageProcessor.from_pretrained("bazudde/potato_model")
model = AutoModelForImageClassification.from_pretrained("bazudde/potato_model")Quick Links
Model Trained Using AutoTrain
- Problem type: Multi-class Classification
- Model ID: 44552112263
- CO2 Emissions (in grams): 0.2586
Validation Metrics
- Loss: 0.098
- Accuracy: 0.923
- Macro F1: 0.911
- Micro F1: 0.923
- Weighted F1: 0.918
- Macro Precision: 0.958
- Micro Precision: 0.923
- Weighted Precision: 0.933
- Macro Recall: 0.889
- Micro Recall: 0.923
- Weighted Recall: 0.923
- Downloads last month
- 5
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="bazudde/potato_model") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")