piper / app.py
aigmixer's picture
Update app.py
791253b
raw
history blame
1.67 kB
import gradio as gr
import wave
import numpy as np
from io import BytesIO
from huggingface_hub import hf_hub_download
from piper import PiperVoice # Adjust import as per your project structure
#file_path = hf_hub_download("rhasspy/piper-voices", "en_GB-alan-medium.onnx")
def synthesize_speech(text):
# Load the PiperVoice model and configuration
# model_path = "en_GB-alan-medium.onnx" # this is for loading local model
# config_path = "en_GB-alan-medium.onnx.json" # for loading local json
model_path = hf_hub_download(repo_id="rhasspy/piper-voices", filename="en/en_GB/alan/medium/en_GB-alan-medium.onnx")
config_path = hf_hub_download(repo_id="rhasspy/piper-voices", filename="en/en_GB/alan/medium/en_GB-alan-medium.onnx.json")
voice = PiperVoice.load(model_path, config_path)
# Create an in-memory buffer for the WAV file
buffer = BytesIO()
with wave.open(buffer, 'wb') as wav_file:
wav_file.setframerate(voice.config.sample_rate)
wav_file.setsampwidth(2) # 16-bit
wav_file.setnchannels(1) # mono
# Synthesize speech
voice.synthesize(text, wav_file)
# Convert buffer to NumPy array for Gradio output
buffer.seek(0)
audio_data = np.frombuffer(buffer.read(), dtype=np.int16)
return audio_data.tobytes()
# Create a Gradio interface with labels
iface = gr.Interface(
fn=synthesize_speech,
inputs=gr.Textbox(label="Input Text"),
outputs=[gr.Audio(label="Synthesized Speech")],
title="Text to Speech Synthesizer",
description="Enter text to synthesize it into speech using PiperVoice.",
allow_flagging="never"
)
# Run the app
iface.launch()