| 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") | |