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()
- Downloads last month
- 25
Inference Providers
NEW
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API:
The model has no library tag.