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