cjones/WhisperLive
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

recieve float buffer

Browse Source
This commit is contained in:
makaveli10
2023-08-09 04:03:28 +08:00
parent fbaa1ce6ab
commit 5a98f28772
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
whisper_live/server.py
+2 -4
View File
@@ -89,10 +89,7 @@ class TranscriptionServer:
while True: while True:
try: try:
frame_data = websocket.recv() frame_data = websocket.recv()
data = json.loads(frame_data) frame_np = np.frombuffer(frame_data, dtype=np.float32)
base64_audio = data["audio"]
binary_audio = base64.b64decode(base64_audio)
frame_np = np.frombuffer(binary_audio, dtype=np.float32)
try: try:
speech_prob = self.vad_model(torch.from_numpy(frame_np.copy()), self.RATE).item() speech_prob = self.vad_model(torch.from_numpy(frame_np.copy()), self.RATE).item()
@@ -117,6 +114,7 @@ class TranscriptionServer:
except Exception as e: except Exception as e:
logging.error(e)
self.clients[websocket].cleanup() self.clients[websocket].cleanup()
self.clients.pop(websocket) self.clients.pop(websocket)
self.clients_start_time.pop(websocket) self.clients_start_time.pop(websocket)