# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("gustproof/dnbr-tagger-preview1", dtype="auto")Quick Links
A lightweight multilabel classifier trained on an arbitrary subset of nyanko7/danbooru2023. Trained for ~40M samples at native resolution ~238x238. Decent performance for its size.
- Downloads last month
- 1
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="gustproof/dnbr-tagger-preview1") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")