Update app.py

app.py

@@ -1,8 +1,9 @@
+import torch
 import gradio as gr
 import librosa 
 from transformers import AutoFeatureExtractor, AutoModelForSeq2SeqLM, AutoTokenizer, pipeline
+from TTS.utils.manage import ModelManager
 
-import torch
 first_generation = True
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
@@ -26,6 +27,11 @@ model_checkpoint = "hackathon-pln-es/es_text_neutralizer"
 tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
 
+
+manager = ModelManager()
+MODEL_NAMES = manager.list_tts_models()
+
+
 def postproc(input_sentence, preds):
     try:
         preds = preds.replace('De el', 'Del').replace('de el', 'del').replace('  ', ' ')