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
- 34
Inference Providers
NEW
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API:
The HF Inference API does not support image-classification models for pytorch library.
Model tree for ozair23/autotrain-w5nk2-rvmqx
Base model
microsoft/resnet-50