padding fixed

mgonzs13 · Jul 30, 2024 · 739c275 · 739c275
1 parent 88bc90d
commit 739c275
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/whisper_ros/whisper_ros/silero_vad_node.py b/whisper_ros/whisper_ros/silero_vad_node.py
@@ -110,9 +110,11 @@ def audio_cb(self, msg: AudioStamped) -> None:
 
             elif self.recording and "end" in speech_dict:
                 self.recording = False
+                self.data.extend(audio_array.tolist())
 
-                if len(audio_array) / msg.audio.info.rate < 1.0:
-                    pad_size = msg.audio.info.rate - len(audio_array)
+                if len(self.data) / msg.audio.info.rate < 1.0:
+                    pad_size = msg.audio.info.chunk + \
+                        msg.audio.info.rate - len(self.data)
                     self.data = self.data + pad_size * [0.0]
 
                 vad_msg = Float32MultiArray()