Fix streaming in webui (#643)

fishaudio · Oct 30, 2024 · 5c7e397 · 5c7e397
1 parent f8a57fb
commit 5c7e397
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tools/webui.py b/tools/webui.py
@@ -146,7 +146,9 @@ def inference(
         segments.append(fake_audios)
 
         if streaming:
-            yield (fake_audios * 32768).astype(np.int16).tobytes(), None, None
+            wav_header = wav_chunk_header()
+            audio_data = (fake_audios * 32768).astype(np.int16).tobytes()
+            yield wav_header + audio_data, None, None
 
     if len(segments) == 0:
         return (