Commit
·
5ae1a45
1
Parent(s):
4467bd1
try to add alignment
Browse files- handler.py +4 -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 |
-
|
|
|
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 |
-
|
|
|
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
|
|