update tts

app.py

@@ -1,9 +1,13 @@
 import gradio as gr
 from nemo.collections.tts.models import FastPitchModel
 from nemo.collections.tts.models import HifiGanModel
+from nemo.collections.tts.models import MixerTTSModel
+
+# spec_generator_2 = MixerTTSModel.from_pretrained("tts_en_lj_mixerttsx")
+# model1 = HifiGanModel.from_pretrained(model_name="tts_en_lj_hifigan_ft_mixerttsx")
 
 spec_generator = FastPitchModel.from_pretrained("tts_en_fastpitch_multispeaker")
-tts_model = HifiGanModel.from_pretrained(model_name="tts_en_hifitts_hifigan_ft_fastpitch")
+voc_model = HifiGanModel.from_pretrained(model_name="tts_en_hifitts_hifigan_ft_fastpitch")
 
 def greet(name):
     return "Hello " + name + "!!"
@@ -12,12 +16,12 @@ def generate_tts(text: str, speaker: int = 0):
     sr = 22050
     parsed = spec_generator.parse(text)
     spectrogram = spec_generator.generate_spectrogram(tokens=parsed, speaker=speaker)
-    audio = tts_model.convert_spectrogram_to_audio(spec=spectrogram)
+    audio = voc_model.convert_spectrogram_to_audio(spec=spectrogram)
     return sr, audio
 
 def run():
     demo = gr.Interface(
-        fn=greet,
+        fn=generate_tts,
         inputs=["text",
                 gr.Slider(0, 10, step=1)],
         outputs="audio",