File size: 613 Bytes
e4d78d5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
from transformers import AutoImageProcessor, AutoModelForImageClassification
from PIL import Image
import requests
processor = AutoImageProcessor.from_pretrained("Hyunil/CSATv2", trust_remote_code=True)
model = AutoModelForImageClassification.from_pretrained("Hyunil/CSATv2", trust_remote_code=True)
url = "https://images.unsplash.com/photo-1516116216624-53e697fedbea"
image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
inputs = processor(image, return_tensors="pt")
outputs = model(**inputs)
probs = outputs.logits.softmax(dim=-1)
top_prob, top_idx = probs.topk(5)
print(top_idx, top_prob) |