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
|
|
|