mestrevh/computer-vision-cifar-10

Vision Transformer Fine-tuned on CIFAR-10

This model is a Vision Transformer (ViT) fine-tuned on the CIFAR-10 dataset for image classification.

Model Description

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.

Intended Use

This model is intended for image classification tasks.

How to Use

from transformers import pipeline

classifier = pipeline('image-classification', model='your-username/your-model-name')
results = classifier('path/to/your/image.jpg')
print(results)