diff --git a/examples/api/main.py b/examples/api/main.py index 1395f2f..d407a5d 100644 --- a/examples/api/main.py +++ b/examples/api/main.py @@ -97,7 +97,9 @@ async def generate_voice(params: ChatTTSParams): # zip all of the audio files together buf = io.BytesIO() - with zipfile.ZipFile(buf, "a", compression=zipfile.ZIP_DEFLATED, allowZip64=False) as f: + with zipfile.ZipFile( + buf, "a", compression=zipfile.ZIP_DEFLATED, allowZip64=False + ) as f: for idx, wav in enumerate(wavs): f.writestr(wav_arr_to_mp3_view(wav), f"{idx}.mp3") logger.info("Audio generation successful.") diff --git a/examples/cmd/stream.py b/examples/cmd/stream.py index 75296a1..dd4b9b3 100644 --- a/examples/cmd/stream.py +++ b/examples/cmd/stream.py @@ -3,12 +3,13 @@ import threading import time import random -import pyaudio # please install it manually +import pyaudio # please install it manually import numpy as np import ChatTTS from tools.audio import batch_unsafe_float_to_int16 + # 流式声音处理器 class AudioStreamer: def __init__(self): diff --git a/tools/audio/np.py b/tools/audio/np.py index 32ba820..527a140 100644 --- a/tools/audio/np.py +++ b/tools/audio/np.py @@ -13,6 +13,7 @@ def unsafe_float_to_int16(audio: np.ndarray) -> np.ndarray: audio16 = audio.astype(np.int16) return audio16 + @jit def batch_unsafe_float_to_int16(audios: list[np.ndarray]) -> list[np.ndarray]: """