public-mirrors/Kokoro-FastAPI
1
0
Fork 0
mirror of https://github.com/remsky/Kokoro-FastAPI.git synced 2025-04-13 09:39:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update audio.py

Browse source
This commit is contained in:
DINMAY KUMAR BRAHMA 2025-01-03 00:28:59 +05:30 • committed by GitHub
parent e9014e620e
commit 8ccca1fcad
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
api/src/services/audio.py
View file

@ -30,7 +30,7 @@ class AudioService:
if output_format == "wav": if output_format == "wav":
logger.info("Writing to WAV format...") logger.info("Writing to WAV format...")
# Ensure audio_data is in int16 format for WAV # Ensure audio_data is in int16 format for WAV
audio_data_wav = audio_data.astype(np.int16) audio_data_wav = (audio_data / np.abs(audio_data).max() * np.iinfo(np.int16).max).astype(np.int16) # Normalize
sf.write(buffer, audio_data_wav, sample_rate, format="WAV") sf.write(buffer, audio_data_wav, sample_rate, format="WAV")
elif output_format == "mp3": elif output_format == "mp3":
logger.info("Converting to MP3 format...") logger.info("Converting to MP3 format...")
@ -45,7 +45,7 @@ class AudioService:
elif output_format == "pcm": elif output_format == "pcm":
logger.info("Extracting PCM data...") logger.info("Extracting PCM data...")
# Ensure audio_data is in int16 format for PCM # Ensure audio_data is in int16 format for PCM
audio_data_pcm = audio_data.astype(np.int16) audio_data_pcm = (audio_data / np.abs(audio_data).max() * np.iinfo(np.int16).max).astype(np.int16) # Normalize
buffer.write(audio_data_pcm.tobytes()) buffer.write(audio_data_pcm.tobytes())
else: else:
raise ValueError( raise ValueError(