from transformers import AutoModelForImageClassification, AutoImageProcessor | |
model = AutoModelForImageClassification.from_pretrained("shravvvv/SAG-ViT", trust_remote_code=True) | |
processor = AutoImageProcessor.from_pretrained("shravvvv/SAG-ViT", trust_remote_code=True) | |
print(model) | |