Upload model

modeling_test.py

@@ -2,7 +2,8 @@ from transformers import PreTrainedModel
 from .configuration_test import TestConfig
 import torch.nn as nn
 from transformers import AutoModelForMaskedLM, AutoConfig
-from transformers import AutoModelForSequenceClassification
+import librosa
+import os
 
 
 class TestModel(PreTrainedModel):
@@ -16,5 +17,9 @@ class TestModel(PreTrainedModel):
             AutoConfig.from_pretrained("albert/albert-base-v2")
         )
 
+    def get_audio_duration(self):
+        arr, sr = librosa.load(os.path.join(os.path.dirname(os.path.abspath(__file__)), "output1.wav"))
+        return librosa.get_duration(arr, sr)
+
     def forward(self, tensor):
         return self.model1(tensor)