mirrors/whispering
1
0
Fork 0
mirror of https://github.com/shirayu/whispering.git synced 2024-05-20 01:08:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix

Browse source
This commit is contained in:
Yuta Hayashibe 2022-10-02 19:48:41 +09:00
parent 936d5d0c45
commit 847eee5819
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
whispering/serve.py
View file

@ -21,7 +21,7 @@ async def serve_with_websocket_main(websocket):
)
while True:
logger.debug(f"Segment #: {idx}")
logger.debug(f"Audio #: {idx}")
try:
message = await websocket.recv()
except ConnectionClosedOK:
@ -32,9 +32,9 @@ async def serve_with_websocket_main(websocket):
continue
logger.debug(f"Message size: {len(message)}")
segment = np.frombuffer(message, dtype=np.float32)
audio = np.frombuffer(message, dtype=np.float32)
for chunk in g_wsp.transcribe(
segment=segment, # type: ignore
audio=audio, # type: ignore
ctx=ctx,
):
await websocket.send(chunk.json())