from transformers import ViTForImageClassification

# Assuming you're loading the model using the ViTForImageClassification class
model = ViTForImageClassification.from_pretrained('/home/user/.cache/huggingface/hub/models--rameye--1/snapshots/c2c5c38a641e6c048f33c2db25afa765727a04ed', from_tf=True)