from transformers import AutoConfig, AutoModel
from modeling_sagvit import SAGViTClassifier

# Initialize config and model
config = AutoConfig.from_pretrained("shravvvv/SAG-ViT")
model = AutoModel.from_pretrained("shravvvv/SAG-ViT", config=config)

# Push model to the Hub
model.push_to_hub("shravvvv/SAG-ViT")