autotrain-w5nk2-rvmqx

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