Dominik Macháček
commited on
Commit
·
827425b
1
Parent(s):
4a89935
small code review
Browse files- whisper_online.py +3 -5
whisper_online.py
CHANGED
@@ -558,7 +558,6 @@ class VACOnlineASRProcessor(OnlineASRProcessor):
|
|
558 |
|
559 |
def insert_audio_chunk(self, audio):
|
560 |
res = self.vac(audio)
|
561 |
-
print(res)
|
562 |
self.audio_buffer = np.append(self.audio_buffer, audio)
|
563 |
|
564 |
if res is not None:
|
@@ -584,11 +583,10 @@ class VACOnlineASRProcessor(OnlineASRProcessor):
|
|
584 |
if self.status == 'voice':
|
585 |
self.online.insert_audio_chunk(self.audio_buffer)
|
586 |
self.current_online_chunk_buffer_size += len(self.audio_buffer)
|
587 |
-
if self.status is not None:
|
588 |
self.clear_buffer()
|
589 |
-
else:
|
590 |
-
# We keep
|
591 |
-
# But
|
592 |
self.buffer_offset += max(0,len(self.audio_buffer)-self.SAMPLING_RATE)
|
593 |
self.audio_buffer = self.audio_buffer[-self.SAMPLING_RATE:]
|
594 |
|
|
|
558 |
|
559 |
def insert_audio_chunk(self, audio):
|
560 |
res = self.vac(audio)
|
|
|
561 |
self.audio_buffer = np.append(self.audio_buffer, audio)
|
562 |
|
563 |
if res is not None:
|
|
|
583 |
if self.status == 'voice':
|
584 |
self.online.insert_audio_chunk(self.audio_buffer)
|
585 |
self.current_online_chunk_buffer_size += len(self.audio_buffer)
|
|
|
586 |
self.clear_buffer()
|
587 |
+
else:
|
588 |
+
# We keep 1 second because VAD may later find start of voice in it.
|
589 |
+
# But we trim it to prevent OOM.
|
590 |
self.buffer_offset += max(0,len(self.audio_buffer)-self.SAMPLING_RATE)
|
591 |
self.audio_buffer = self.audio_buffer[-self.SAMPLING_RATE:]
|
592 |
|