Update model_hf.py

model_hf.py

@@ -8,6 +8,7 @@ import torch.nn.functional as F
 from torch import Tensor
 import fairseq
 from .config_ssl import SSLConfig
+from huggingface_hub import hf_hub_download
 
 ___author__ = "Hemlata Tak"
 __email__ = "tak@eurecom.fr"
@@ -20,8 +21,9 @@ __email__ = "tak@eurecom.fr"
 class SSLModel(nn.Module):
     def __init__(self,device):
         super(SSLModel, self).__init__()
-        
-        cp_path = './xlsr2_300m.pt'   # Change the pre-trained XLSR model path. 
+        repo_id = 'ash56/ssl-aasist'
+        fname = 'xlsr2_300m.pt'
+        cp_path = hf_hub_download(repo_id=repo_id, filename=fname)   # Change the pre-trained XLSR model path. 
         model, cfg, task = fairseq.checkpoint_utils.load_model_ensemble_and_task([cp_path])
         self.model = model[0]
         self.model_device=device