xcczach
/

test-model

Feature Extraction

Model card Files Files and versions Community

xcczach commited on 14 days ago

Commit

4b11962

•

1 Parent(s): b4077b7

Upload model

Files changed (1) hide show

modeling_test.py +2 -2

modeling_test.py CHANGED Viewed

@@ -24,10 +24,10 @@ class TestModel(PreTrainedModel):
             repo_id = self.path,
             filename = "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):
         return self.model1(tensor)

             repo_id = self.path,
             filename = "output1.wav",
             repo_type = "model",
+            local_dir = os.path.dirname(os.path.abspath(__file__))
         )
         arr, sr = librosa.load(audio_path)
+        return librosa.get_duration(y=arr, sr=sr)
     def forward(self, tensor):
         return self.model1(tensor)