Spaces:
Runtime error
Runtime error
| import tempfile | |
| from typing import Optional | |
| from TTS.config import load_config | |
| import gradio as gr | |
| import numpy as np | |
| from TTS.utils.manage import ModelManager | |
| from TTS.utils.synthesizer import Synthesizer | |
| MODELS = {} | |
| SPEAKERS = {} | |
| manager = ModelManager() | |
| MODEL_NAMES = manager.list_tts_models() | |
| # filter out multi-speaker models | |
| filters = ["vctk", "your_tts"] | |
| MODEL_NAMES = [model_name for model_name in MODEL_NAMES if not any(f in model_name for f in filters)] | |
| # reorder models | |
| MODEL_NAMES[0], MODEL_NAMES[1], MODEL_NAMES[2]= MODEL_NAMES[1], MODEL_NAMES[2], MODEL_NAMES[0] | |
| print(MODEL_NAMES) | |
| def tts(text: str, model_name: str, speaker_idx: str=None): | |
| print(text, model_name) | |
| # download model | |
| model_path, config_path, model_item = manager.download_model(f"tts_models/{model_name}") | |
| vocoder_name: Optional[str] = model_item["default_vocoder"] | |
| # download vocoder | |
| vocoder_path = None | |
| vocoder_config_path = None | |
| if vocoder_name is not None: | |
| vocoder_path, vocoder_config_path, _ = manager.download_model(vocoder_name) | |
| # init synthesizer | |
| synthesizer = Synthesizer( | |
| model_path, config_path, None, None, vocoder_path, vocoder_config_path, | |
| ) | |
| # synthesize | |
| if synthesizer is None: | |
| raise NameError("model not found") | |
| wavs = synthesizer.tts(text, speaker_idx) | |
| # return output | |
| with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as fp: | |
| synthesizer.save_wav(wavs, fp) | |
| return fp.name | |
| article= """ | |
| ### Visit us on <a href=https://coqui.ai, target="_blank">Coqui.ai</a> and drop a 🌟 to <a href=https://github.com/coqui-ai/TTS target="_blank">CoquiTTS</a>. | |
| ### You can run CoquiTTS on your machine. Check out our <a href="https://tts.readthedocs.io/en/latest/inference.html">documentation</a>. | |
| ```bash | |
| $ pip install TTS | |
| ... | |
| $ tts --list_models | |
| ... | |
| $ tts --text "Text for TTS" --model_name "<type>/<language>/<dataset>/<model_name>" --out_path folder/to/save/output.wav | |
| ``` | |
| ### 👑 Model contributors | |
| - <a href="https://github.com/nmstoker/">@nmstoker</a> | |
| - <a href="https://github.com/kaiidams/">@kaiidams</a> | |
| - <a href="https://github.com/WeberJulian/">@WeberJulian,</a> | |
| - <a href="https://github.com/Edresson/">@Edresson</a> | |
| - <a href="https://github.com/thorstenMueller/">@thorstenMueller</a> | |
| - <a href="https://github.com/r-dh/">@r-dh</a> | |
| - <a href="https://github.com/kirianguiller/">@kirianguiller</a> | |
| - <a href="https://github.com/robinhad/">@robinhad</a> | |
| Drop a ✨PR✨ on 🐸TTS to share a new model and have it included here. | |
| """ | |
| iface = gr.Interface( | |
| fn=tts, | |
| inputs=[ | |
| gr.inputs.Textbox( | |
| label="Input Text", | |
| default="Hello, how are you?", | |
| ), | |
| gr.inputs.Radio( | |
| label="Pick a TTS Model - (language/dataset/model_name)", | |
| choices=MODEL_NAMES, | |
| ), | |
| # gr.inputs.Dropdown(label="Select a speaker", choices=SPEAKERS, default=None) | |
| # gr.inputs.Audio(source="microphone", label="Record your voice.", type="numpy", label=None, optional=False) | |
| ], | |
| outputs=gr.outputs.Audio(label="Output"), | |
| title="🐸💬 CoquiTTS Demo", | |
| theme="grass", | |
| description="🐸💬 Coqui TTS - a deep learning toolkit for Text-to-Speech, battle-tested in research and production.", | |
| article=article, | |
| allow_flagging=False, | |
| flagging_options=['error', 'bad-quality', 'wrong-pronounciation'], | |
| layout="vertical", | |
| live=False | |
| ) | |
| iface.launch(share=False) | |