| 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) |