Upload model

modeling_test.py

@@ -3,6 +3,7 @@ from .configuration_test import TestConfig
 import torch.nn as nn
 from transformers import AutoModelForMaskedLM, AutoConfig
 import librosa
+from huggingface_hub import hf_hub_download
 import os
 
 
@@ -16,9 +17,16 @@ class TestModel(PreTrainedModel):
         self.model2 = AutoModelForMaskedLM.from_config(
             AutoConfig.from_pretrained("albert/albert-base-v2")
         )
+        self.path = config.name_or_path
 
     def get_audio_duration(self):
-        arr, sr = librosa.load(os.path.join(os.path.dirname(os.path.abspath(__file__)), "output1.wav"))
+        audio_path = hf_hub_download(
+            repo_id = self.path,
+            file_path = "output1.wav",
+            repo_type = "model",
+            # local_dir = os.path.abspath(os.path.dirname(os.path.abspath(__file__)))
+        )
+        arr, sr = librosa.load(audio_path)
         return librosa.get_duration(arr, sr)
 
     def forward(self, tensor):