Modelpublic

zkunikowska/cats-vs-dogs-deit-tiny

sourceHugging Faceupdated 11mo agoView on Hugging Face

0likes5downloads

Model Card

Cats vs Dogs — DeiT Tiny (fine-tuned)

Base: facebook/deit-tiny-patch16-224 Task: image-classification Labels: ['cat', 'dog']

python

from transformers import pipeline
clf = pipeline("image-classification", model="zkunikowska/cats-vs-dogs-deit-tiny")
clf("dog_or_cat.jpg")