Create the TTS engine using Fastspeech

tts.py

@@ -0,0 +1,17 @@
+from fairseq.checkpoint_utils import load_model_ensemble_and_task_from_hf_hub
+from fairseq.models.text_to_speech.hub_interface import TTSHubInterface
+
+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)
+
+def tts(text):
+  text = "Hello, this is a test run."
+  
+  sample = TTSHubInterface.get_model_input(task, text)
+  wav, rate = TTSHubInterface.get_prediction(task, model, generator, sample)
+  return wave, rate