Fix VAD syntax & add vad handling case

jhj0517 · Oct 30, 2024 · eec0c16 · eec0c16
1 parent ddbe0b6
commit eec0c16
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/modules/whisper/base_transcription_pipeline.py b/modules/whisper/base_transcription_pipeline.py
@@ -135,12 +135,17 @@ def run(self,
                 speech_pad_ms=vad_params.speech_pad_ms
             )
 
-            audio, speech_chunks = self.vad.run(
+            vad_processed, speech_chunks = self.vad.run(
                 audio=audio,
                 vad_parameters=vad_options,
                 progress=progress
             )
 
+            if vad_processed.size > 0:
+                audio = vad_processed
+            else:
+                vad_params.vad_filter = False
+
         result, elapsed_time = self.transcribe(
             audio,
             progress,
@@ -150,7 +155,7 @@ def run(self,
         if vad_params.vad_filter:
             result = self.vad.restore_speech_timestamps(
                 segments=result,
-                speech_chunks=vad_params.speech_chunks,
+                speech_chunks=speech_chunks,
             )
 
         if diarization_params.is_diarize: