| import streamlit as st | |
| from fairseq.models.text_to_speech.hub_interface import TTSHubInterface | |
| import IPython.display as ipd | |
| models, cfg, task = load_model_ensemble_and_task_from_hf_hub( | |
| "facebook/fastspeech2-en-ljspeech", | |
| arg_overrides={"vocoder": "hifigan", "fp16": False} | |
| ) | |
| model = models[0] | |
| TTSHubInterface.update_cfg_with_data_cfg(cfg, task.data_cfg) | |
| generator = task.build_generator(model, cfg) | |
| text = "Hello, this is a test run." | |
| sample = TTSHubInterface.get_model_input(task, text) | |
| wav, rate = TTSHubInterface.get_prediction(task, model, generator, sample) | |
| ipd.Audio(wav, rate=rate) |