From 2300eedc8bb1fe0a397d137827937cfd4c12206a Mon Sep 17 00:00:00 2001 From: makaveli10 Date: Tue, 9 Jan 2024 13:35:38 +0800 Subject: [PATCH] fix: error messages --- whisper_live/server.py | 11 ++++------- whisper_live/transcriber.py | 3 --- 2 files changed, 4 insertions(+), 10 deletions(-) diff --git a/whisper_live/server.py b/whisper_live/server.py index 81ca9f3..b019af3 100644 --- a/whisper_live/server.py +++ b/whisper_live/server.py @@ -120,7 +120,7 @@ class TranscriptionServer: elapsed_time = time.time() - self.clients_start_time[websocket] if elapsed_time >= self.max_connection_time: self.clients[websocket].disconnect() - logging.warning(f"{self.clients[websocket]} Client disconnected due to overtime.") + logging.warning(f"Client with uid '{self.clients[websocket].client_uid}' disconnected due to overtime.") self.clients[websocket].cleanup() self.clients.pop(websocket) self.clients_start_time.pop(websocket) @@ -129,13 +129,11 @@ class TranscriptionServer: break except Exception as e: - logging.error(e) + logging.info(f"[ERROR]: Client with uid '{self.clients[websocket].client_uid}' Disconnected.") if self.clients[websocket].model_size is not None: self.clients[websocket].cleanup() self.clients.pop(websocket) self.clients_start_time.pop(websocket) - logging.info("Connection Closed.") - logging.info(self.clients) del websocket break @@ -407,10 +405,10 @@ class ServeClient: }) ) except Exception as e: - logging.error(f"[ERROR]: {e}") + logging.error(f"[ERROR]: Failed to send message to client: {e}") except Exception as e: - logging.error(f"[ERROR]: {e}") + logging.error(f"[ERROR]: Failed to transcribe audio chunk: {e}") time.sleep(0.01) def format_segment(self, start, end, text): @@ -518,4 +516,3 @@ class ServeClient: """ logging.info("Cleaning up.") self.exit = True - self.transcriber.destroy() diff --git a/whisper_live/transcriber.py b/whisper_live/transcriber.py index cf301d3..e6f2f1c 100644 --- a/whisper_live/transcriber.py +++ b/whisper_live/transcriber.py @@ -934,9 +934,6 @@ class WhisperModel: ) ] - def destroy(self): - del self.model - def restore_speech_timestamps( segments: Iterable[Segment],