Image Classification
Transformers
PyTorch
beit
Trained with AutoTrain
vision
How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("image-classification", model="rafferty/autotrain-amber-mine-tutorial")
pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")
# Load model directly
from transformers import AutoImageProcessor, AutoModelForImageClassification

processor = AutoImageProcessor.from_pretrained("rafferty/autotrain-amber-mine-tutorial")
model = AutoModelForImageClassification.from_pretrained("rafferty/autotrain-amber-mine-tutorial")
Quick Links

Model Trained Using AutoTrain

  • Problem type: Binary Classification
  • Model ID: 49296119123
  • CO2 Emissions (in grams): 0.3929

Validation Metrics

  • Loss: 0.107
  • Accuracy: 0.980
  • Precision: 0.962
  • Recall: 1.000
  • AUC: 0.995
  • F1: 0.980
Downloads last month
4
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

Spaces using rafferty/autotrain-amber-mine-tutorial 2