Dominik Macháček commited on
Commit
827425b
·
1 Parent(s): 4a89935

small code review

Browse files
Files changed (1) hide show
  1. 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: # we are at the beginning of process, no voice has ever been detected
590
- # We keep the 1s because VAD may later find start of voice in it.
591
- # But trimming it to prevent OOM.
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