How to use from the
Use from the
Transformers library
# Load model directly
from transformers import AutoImageProcessor, ViTForSketchClassification

processor = AutoImageProcessor.from_pretrained("WinKawaks/SketchXAI-Tiny-QuickDraw345")
model = ViTForSketchClassification.from_pretrained("WinKawaks/SketchXAI-Tiny-QuickDraw345", device_map="auto")
Quick Links

A full description of this project can be found at https://sketchxai.github.io/.

Downloads last month
18
Safetensors
Model size
6.12M params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Dataset used to train WinKawaks/SketchXAI-Tiny-QuickDraw345