From b37d3cafb3f7de79c80c8c4d7caa34ab560a31aa Mon Sep 17 00:00:00 2001 From: Quentin Fuxa Date: Mon, 3 Mar 2025 10:34:32 +0100 Subject: [PATCH] ffmpeg timout from 5 to 15s; diarization lag does not stay = 0 at the beginning --- whisper_fastapi_online_server.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/whisper_fastapi_online_server.py b/whisper_fastapi_online_server.py index 0ba9844..c4a12f4 100644 --- a/whisper_fastapi_online_server.py +++ b/whisper_fastapi_online_server.py @@ -120,8 +120,7 @@ class SharedState: remaining_time_transcription = max(0, round(current_time - self.beg_loop - self.end_buffer, 2)) # Calculate remaining time for diarization - if self.end_attributed_speaker > 0: - remaining_time_diarization = max(0, round(max(self.end_buffer, self.tokens[-1].end if self.tokens else 0) - self.end_attributed_speaker, 2)) + remaining_time_diarization = max(0, round(max(self.end_buffer, self.tokens[-1].end if self.tokens else 0) - self.end_attributed_speaker, 2)) return { "tokens": self.tokens.copy(), @@ -420,7 +419,7 @@ async def websocket_endpoint(websocket: WebSocket): loop.run_in_executor( None, ffmpeg_process.stdout.read, ffmpeg_buffer_from_duration ), - timeout=5.0 + timeout=15.0 ) except asyncio.TimeoutError: logger.warning("FFmpeg read timeout. Restarting...")