Views
No views yet
1from transformers import AutoImageProcessor, AutoModelForImageClassification
2from PIL import Image
3import torch
4
5processor = AutoImageProcessor.from_pretrained("myneighborh/vehicle-model-classifier")
6model = AutoModelForImageClassification.from_pretrained("myneighborh/vehicle-model-classifier")
7
8image = Image.open("path_to_image.jpg")
9inputs = processor(images=image, return_tensors="pt")
10outputs = model(**inputs)
11pred = outputs.logits.argmax(-1).item()
12label = model.config.id2label[pred]
13print("Predicted label:", label)