ericmattmann commited on
Commit
5ae1a45
·
1 Parent(s): 4467bd1

try to add alignment

Browse files
Files changed (2) hide show
  1. handler.py +4 -2
  2. test endpoint.ipynb +0 -0
handler.py CHANGED
@@ -289,7 +289,8 @@ class EndpointHandler:
289
  del model_a
290
  if info:
291
  print(transcription["segments"][0:10000])
292
- logger.info(transcription["segments"][0:10000])
 
293
 
294
  et = time.time()
295
  elapsed_time = et - st
@@ -309,7 +310,8 @@ class EndpointHandler:
309
  diarize_segments = self.diarize_model(audio_nparray, min_speakers=min_speakers, max_speakers=max_speakers)
310
  if info:
311
  print(diarize_segments)
312
- logger.info(diarize_segments)
 
313
 
314
  transcription = whisperx.assign_word_speakers(diarize_segments, transcription)
315
 
 
289
  del model_a
290
  if info:
291
  print(transcription["segments"][0:10000])
292
+ else:
293
+ logger.info(transcription["segments"][0:1_000])
294
 
295
  et = time.time()
296
  elapsed_time = et - st
 
310
  diarize_segments = self.diarize_model(audio_nparray, min_speakers=min_speakers, max_speakers=max_speakers)
311
  if info:
312
  print(diarize_segments)
313
+ else:
314
+ logger.info(diarize_segments)
315
 
316
  transcription = whisperx.assign_word_speakers(diarize_segments, transcription)
317
 
test endpoint.ipynb CHANGED
The diff for this file is too large to render. See raw diff