Spaces:
Runtime error
Runtime error
Commit
Β·
226405b
1
Parent(s):
346f0e5
Update app.py
Browse files
app.py
CHANGED
|
@@ -20,12 +20,6 @@ fastspeech2 = TFAutoModel.from_pretrained("tensorspeech/tts-fastspeech2-ljspeech
|
|
| 20 |
tacotron2 = TFAutoModel.from_pretrained("tensorspeech/tts-tacotron2-ljspeech-en", name="tacotron2")
|
| 21 |
melgan = TFAutoModel.from_pretrained("tensorspeech/tts-melgan-ljspeech-en", name="melgan")
|
| 22 |
mb_melgan = TFAutoModel.from_pretrained("tensorspeech/tts-mb_melgan-ljspeech-en", name="mb_melgan")
|
| 23 |
-
melgan_stft_config = AutoConfig.from_pretrained('TensorFlowTTS/examples/melgan_stft/conf/melgan_stft.v1.yaml')
|
| 24 |
-
melgan_stft = TFAutoModel.from_pretrained(
|
| 25 |
-
config=melgan_stft_config,
|
| 26 |
-
pretrained_path="melgan.stft-2M.h5",
|
| 27 |
-
name="melgan_stft"
|
| 28 |
-
)
|
| 29 |
|
| 30 |
MODEL_DICT = {
|
| 31 |
"Fastspeech2" : fastspeech2,
|
|
@@ -67,7 +61,7 @@ def inference(input):
|
|
| 67 |
raise ValueError("Only TACOTRON, FASTSPEECH, FASTSPEECH2 are supported on text2mel_name")
|
| 68 |
|
| 69 |
# vocoder part
|
| 70 |
-
if vocoder_name == "Melgan"
|
| 71 |
audio = vocoder_model(mel_outputs)[0, :, 0]
|
| 72 |
elif vocoder_name == "MB-Melgan":
|
| 73 |
audio = vocoder_model(mel_outputs)[0, :, 0]
|
|
|
|
| 20 |
tacotron2 = TFAutoModel.from_pretrained("tensorspeech/tts-tacotron2-ljspeech-en", name="tacotron2")
|
| 21 |
melgan = TFAutoModel.from_pretrained("tensorspeech/tts-melgan-ljspeech-en", name="melgan")
|
| 22 |
mb_melgan = TFAutoModel.from_pretrained("tensorspeech/tts-mb_melgan-ljspeech-en", name="mb_melgan")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
|
| 24 |
MODEL_DICT = {
|
| 25 |
"Fastspeech2" : fastspeech2,
|
|
|
|
| 61 |
raise ValueError("Only TACOTRON, FASTSPEECH, FASTSPEECH2 are supported on text2mel_name")
|
| 62 |
|
| 63 |
# vocoder part
|
| 64 |
+
if vocoder_name == "Melgan":
|
| 65 |
audio = vocoder_model(mel_outputs)[0, :, 0]
|
| 66 |
elif vocoder_name == "MB-Melgan":
|
| 67 |
audio = vocoder_model(mel_outputs)[0, :, 0]
|