Model Trained Using AutoTrain
- Problem type: Image Classification
Validation Metrics
No validation metrics available
#Inference Pipeline
-Here is how to use this model to classify an image of the COCO 2017 dataset into one of the 1,000 ImageNet classes:
from transformers import AutoModelForImageClassification, AutoProcessor
model = AutoModelForImageClassification.from_pretrained("ozair23/autotrain-w5nk2-rvmqx")
processor = AutoProcessor.from_pretrained("ozair23/autotrain-w5nk2-rvmqx")
def predict(image):
inputs = processor(images=image, return_tensors="pt")
outputs = model(**inputs)
return outputs
- Downloads last month
- 32
Inference Providers
NEW
This model is not currently available via any of the supported third-party Inference Providers, and
the HF Inference API does not support pytorch models with pipeline type image-classification
Model tree for ozair23/autotrain-w5nk2-rvmqx
Base model
microsoft/resnet-50