Update app.py

app.py

@@ -13,6 +13,8 @@ import gradio as gr
 MODEL_NAMES = [
     "Fastspeech2 + Melgan", 
     "Tacotron2 + Melgan",
+    "Tacotron2 + MB-Melgan",
+    "Fastspeech2 + MB-Melgan"
 ]
 
 fastspeech = TFAutoModel.from_pretrained("tensorspeech/tts-fastspeech-ljspeech-en", name="fastspeech")
@@ -34,7 +36,7 @@ def inference(input_text, model_type):
     processor = AutoProcessor.from_pretrained("tensorspeech/tts-tacotron2-ljspeech-en")
     input_ids = processor.text_to_sequence(input_text)
     
-    if text2mel_name == "Tacotron":
+    if text2mel_name == "Tacotron2":
         _, mel_outputs, stop_token_prediction, alignment_history = text2mel_model.inference(
             tf.expand_dims(tf.convert_to_tensor(input_ids, dtype=tf.int32), 0),
             tf.convert_to_tensor([len(input_ids)], tf.int32),