Transformers How to use joshjrreynaldo/vit_based_pets_classifier with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("image-feature-extraction", model="joshjrreynaldo/vit_based_pets_classifier") # Load model directly
from transformers import AutoImageProcessor, AutoModel
processor = AutoImageProcessor.from_pretrained("joshjrreynaldo/vit_based_pets_classifier")
model = AutoModel.from_pretrained("joshjrreynaldo/vit_based_pets_classifier", device_map="auto") 
