mcxzfk commited on
Commit
64927cb
1 Parent(s): 2314179

Update src/vad.py

Browse files
Files changed (1) hide show
  1. src/vad.py +4 -2
src/vad.py CHANGED
@@ -174,7 +174,8 @@ class AbstractTranscription(ABC):
174
  segment_index = config.initial_segment_index
175
 
176
  # For each time segment, run whisper
177
- for segment in merged:
 
178
  segment_index += 1
179
  segment_start = segment['start']
180
  segment_end = segment['end']
@@ -194,7 +195,8 @@ class AbstractTranscription(ABC):
194
  # Detected language
195
  detected_language = languageCounter.most_common(1)[0][0] if len(languageCounter) > 0 else None
196
 
197
- print("Running whisper from ", format_timestamp(segment_start), " to ", format_timestamp(segment_end), ", duration: ",
 
198
  segment_duration, "expanded: ", segment_expand_amount, "prompt: ", segment_prompt, "language: ", detected_language)
199
  segment_result = whisperCallable.invoke(segment_audio, segment_index, segment_prompt, detected_language)
200
 
 
174
  segment_index = config.initial_segment_index
175
 
176
  # For each time segment, run whisper
177
+ from tqdm import tqdm
178
+ for i,segment in enumerate(tqdm(merged)):
179
  segment_index += 1
180
  segment_start = segment['start']
181
  segment_end = segment['end']
 
195
  # Detected language
196
  detected_language = languageCounter.most_common(1)[0][0] if len(languageCounter) > 0 else None
197
 
198
+ if i%10 ==0:
199
+ print("Running whisper from ", format_timestamp(segment_start), " to ", format_timestamp(segment_end), ", duration: ",
200
  segment_duration, "expanded: ", segment_expand_amount, "prompt: ", segment_prompt, "language: ", detected_language)
201
  segment_result = whisperCallable.invoke(segment_audio, segment_index, segment_prompt, detected_language)
202