update tts

app.py

@@ -1,13 +1,25 @@
 import gradio as gr
+from nemo.collections.tts.models import FastPitchModel
+from nemo.collections.tts.models import HifiGanModel
+
+spec_generator = FastPitchModel.from_pretrained("tts_en_fastpitch_multispeaker")
+tts_model = HifiGanModel.from_pretrained(model_name="tts_en_hifitts_hifigan_ft_fastpitch")
 
 def greet(name):
     return "Hello " + name + "!!"
 
+def generate_tts(text: str, speaker: int = 0):
+    parsed = spec_generator.parse(text)
+    spectrogram = spec_generator.generate_spectrogram(tokens=parsed, speaker=speaker)
+    audio = tts_model.convert_spectrogram_to_audio(spec=spectrogram)
+    return audio
+
 def run():
     demo = gr.Interface(
         fn=greet,
-        inputs="text",
-        outputs="text",
+        inputs=["text",
+                gr.Slider(4, 6, step=1)],
+        outputs="audio",
     )
 
     demo.launch(server_name="0.0.0.0", server_port=7860)