Update custom_interface_app.py

custom_interface_app.py

@@ -172,17 +172,17 @@ class ASR(Pretrained):
         
 
 
-    def classify_file_whisper_mkd(self, path, device):
+    def classify_file_whisper_mkd(self, waveform, device):
         # Load the audio file
-        waveform, sr = librosa.load(path, sr=16000)
+        # waveform, sr = librosa.load(path, sr=16000)
 
         # Get audio length in seconds
-        audio_length = len(waveform) / sr
+        audio_length = len(waveform) / 16000
         
         if audio_length >= 20:
             # split audio every 20 seconds
             segments = []
-            max_duration = 20 * sr  # Maximum segment duration in samples (20 seconds)
+            max_duration = 20 * 16000  # Maximum segment duration in samples (20 seconds)
             num_segments = int(np.ceil(len(waveform) / max_duration))
             start = 0
             for i in range(num_segments):
@@ -190,7 +190,7 @@ class ASR(Pretrained):
                 if end > len(waveform):
                     end = len(waveform)
                 segment_part = waveform[start:end]
-                segment_len = len(segment_part) / sr
+                segment_len = len(segment_part) / 16000
                 if segment_len < 1:
                     continue
                 segments.append(segment_part)