How to use parneetsingh022/dog-cat-classification with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("parneetsingh022/dog-cat-classification", device_map="auto")
