diff --git a/whisper_streaming/serve.py b/whisper_streaming/serve.py index e2c9217..d452fe1 100644 --- a/whisper_streaming/serve.py +++ b/whisper_streaming/serve.py @@ -26,8 +26,8 @@ async def serve_with_websocket_main(websocket): logger.debug(f"Message size: {len(message)}") segment = np.frombuffer(message, dtype=np.float32) - chunks = [chunk.dict() for chunk in g_wsp.transcribe(segment=segment)] - await websocket.send(json.dumps(chunks)) + for chunk in g_wsp.transcribe(segment=segment): + await websocket.send(chunk.json()) idx += 1