Added PyTorchModelHubMixin for HF

megaloc_model.py

@@ -8,9 +8,10 @@ import torch
 import torch.nn as nn
 import torch.nn.functional as F
 import torchvision.transforms as tfm
+from huggingface_hub import PyTorchModelHubMixin
 
 
-class MegaLocModel(nn.Module):
+class MegaLocModel(nn.Module, PyTorchModelHubMixin):
     def __init__(
         self,
         feat_dim=8448,