# Load model directly
from transformers import AutoImageProcessor, AutoModelForImageClassification
processor = AutoImageProcessor.from_pretrained("Enayat/ViT")
model = AutoModelForImageClassification.from_pretrained("Enayat/ViT")Quick Links
ViT
Created a Simple Image Classifier using ViT Transformers, custom trained on blur-dataset with an acc of about 93%. The Dataset link : https://www.kaggle.com/datasets/kwentar/blur-dataset
Architecture
Example Images
defocused_blurred
motion_blurred
sharp
- Downloads last month
- 4
Evaluation results
- Accuracyself-reported0.930
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="Enayat/ViT") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")