| ### How to use | |
| Here is how to use this model to classify an image of the COCO 2017 dataset into one of the 1,000 ImageNet classes: | |
| ```python | |
| from transformers import AutoFeatureExtractor, CvtForImageClassification | |
| from PIL import Image | |
| import requests | |
| url = 'http://images.cocodataset.org/val2017/000000039769.jpg' | |
| image = Image.open(requests.get(url, stream=True).raw) | |
| feature_extractor = AutoFeatureExtractor.from_pretrained('anugunj/cvt-13-384-22k') | |
| model = CvtForImageClassification.from_pretrained('anugunj/cvt-13-384-22k') | |
| inputs = feature_extractor(images=image, return_tensors="pt") | |
| outputs = model(**inputs) | |
| logits = outputs.logits | |
| # model predicts one of the 1000 ImageNet classes | |
| predicted_class_idx = logits.argmax(-1).item() | |
| print("Predicted class:", model.config.id2label[predicted_class_idx]) | |
| ``` |