| import tempfile | |
| import gradio as gr | |
| from piper import PiperVoice | |
| from pathlib import Path | |
| import wave | |
| _FILE = Path(__file__) | |
| _DIR = _FILE.parent | |
| voice = PiperVoice.load( | |
| model_path=_DIR / 'models/rmedium.onnx', | |
| config_path=_DIR / 'models/rmedium.onnx.json' | |
| ) | |
| synthesize_args = {} | |
| def tts(text: str): | |
| print(text) | |
| with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as fp: | |
| with wave.Wave_write(fp) as wav: | |
| voice.synthesize(text, wav, **synthesize_args) | |
| return fp.name | |
| inputs = [gr.Textbox(label="Input", value="Salut, numele meu este Bogdan.", max_lines=10)] | |
| outputs = gr.Audio(label="Output") | |
| demo = gr.Interface(fn=tts, inputs=inputs, outputs=outputs) | |
| demo.launch() | |