google/quickdraw
Updated • 372 • 26
How to use WinKawaks/SketchXAI-Tiny-QuickDraw345 with Transformers:
# Load model directly
from transformers import AutoImageProcessor, ViTForSketchClassification
processor = AutoImageProcessor.from_pretrained("WinKawaks/SketchXAI-Tiny-QuickDraw345")
model = ViTForSketchClassification.from_pretrained("WinKawaks/SketchXAI-Tiny-QuickDraw345")A full description of this project can be found at https://sketchxai.github.io/.