d54f213
1
2
3
4
5
6
7
8
9
10
from PIL import Image import torch image = Image.open("earth.jpg") inputs = processor(images=image, return_tensors="pt") outputs = model(**inputs) pred = outputs.logits.argmax(-1).item() print(model.config.id2label[pred])
