diff --git a/app.py b/app.py new file mode 100644 index 0000000..b7d2650 --- /dev/null +++ b/app.py @@ -0,0 +1,38 @@ +# WebUI by mrfakename +# Demo also available on HF Spaces: https://huggingface.co/spaces/mrfakename/MeloTTS +import gradio as gr +import os, torch, io +# os.system('python -m unidic download') +print("Make sure you've downloaded unidic (python -m unidic download) for this WebUI to work.") +from melo.api import TTS +speed = 1.0 +import tempfile +device = 'cuda' if torch.cuda.is_available() else 'cpu' +models = { + 'EN': TTS(language='EN', device=device), + 'ES': TTS(language='ES', device=device), + 'FR': TTS(language='FR', device=device), + 'ZH': TTS(language='ZH', device=device), + 'JP': TTS(language='JP', device=device), + 'KR': TTS(language='KR', device=device), +} +speaker_ids = models['EN'].hps.data.spk2id +def synthesize(speaker, text, speed, language, progress=gr.Progress()): + bio = io.BytesIO() + models[language].tts_to_file(text, models[language].hps.data.spk2id[speaker], bio, speed=speed, pbar=progress.tqdm, format='wav') + return bio.getvalue() +def load_speakers(language): + return gr.update(value=list(models[language].hps.data.spk2id.keys())[0], choices=list(models[language].hps.data.spk2id.keys())) +with gr.Blocks() as demo: + gr.Markdown('# MeloTTS WebUI\n\nA WebUI for MeloTTS.') + with gr.Group(): + speaker = gr.Dropdown(speaker_ids.keys(), interactive=True, value='EN-Default', label='Speaker') + language = gr.Radio(['EN', 'ES', 'FR', 'ZH', 'JP', 'KR'], label='Language', value='EN') + language.input(load_speakers, inputs=language, outputs=speaker) + speed = gr.Slider(label='Speed', minimum=0.1, maximum=10.0, value=1.0, interactive=True, step=0.1) + text = gr.Textbox(label="Text to speak", value='The field of text to speech has seen rapid development recently') + btn = gr.Button('Synthesize', variant='primary') + aud = gr.Audio(interactive=False) + btn.click(synthesize, inputs=[speaker, text, speed, language], outputs=[aud]) + gr.Markdown('WebUI by [mrfakename](https://twitter.com/realmrfakename).') +demo.queue(api_open=False, default_concurrency_limit=10).launch(show_api=False)