| license: mit | |
| lat_mean=39.951163 lat_std=0.000596 lon_mean=-75.19156 lon_std=0.000609 | |
| class CustomViTModel(ViTForImageClassification, PyTorchModelHubMixin): | |
| def __init__(self, config): | |
| super().__init__(config) | |
| self.classifier = nn.Linear(self.classifier.in_features, config.num_labels) | |
| model = CustomViTModel.from_pretrained("img2gpspenn/vit") | |
| model.eval() | |