Spaces:
Runtime error
Runtime error
Commit
Β·
70e4553
1
Parent(s):
00c8d59
Update app.py
Browse files
app.py
CHANGED
|
@@ -28,12 +28,10 @@ MODEL_DICT = {
|
|
| 28 |
"MB-Melgan": mb_melgan,
|
| 29 |
}
|
| 30 |
|
| 31 |
-
def inference(
|
| 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(
|
| 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":
|