This model is a Vision Transformer (ViT) fine-tuned on the CIFAR-10 dataset for image classification.
The model was trained on the CIFAR-10 dataset, which consists of 60,000 32x32 color images in 10 classes, with 6,000 images per class. The fine-tuning was performed on the pretrained ViT base model.
This model is intended for image classification tasks.
1from transformers import pipeline
2
3classifier = pipeline('image-classification', model='your-username/your-model-name')
4results = classifier('path/to/your/image.jpg')
5print(results)
