StevenLimcorn commited on
Commit
70e4553
β€’
1 Parent(s): 00c8d59

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -28,12 +28,10 @@ MODEL_DICT = {
28
  "MB-Melgan": mb_melgan,
29
  }
30
 
31
- def inference(input):
32
- input_text, model_type = input[0], input[1]
33
-
34
  text2mel_name, vocoder_name = model_type.split(" + ")
35
  text2mel_model, vocoder_model = MODEL_DICT[text2mel_name], MODEL_DICT[vocoder_name]
36
- processor = AutoProcessor.from_pretrained(text2mel_name)
37
  input_ids = processor.text_to_sequence(input_text)
38
 
39
  if text2mel_name == "Tacotron":
 
28
  "MB-Melgan": mb_melgan,
29
  }
30
 
31
+ def inference(input_text, model_type):
 
 
32
  text2mel_name, vocoder_name = model_type.split(" + ")
33
  text2mel_model, vocoder_model = MODEL_DICT[text2mel_name], MODEL_DICT[vocoder_name]
34
+ processor = AutoProcessor.from_pretrained("tensorspeech/tts-tacotron2-ljspeech-en")
35
  input_ids = processor.text_to_sequence(input_text)
36
 
37
  if text2mel_name == "Tacotron":