Update app.py

app.py

@@ -14,20 +14,21 @@ from inference import load_tacotron_model, load_wavegru_net, mel_to_wav, text_to
 from wavegru_cpp import extract_weight_mask, load_wavegru_cpp
 
 
-def speak(text):
-    alphabet, tacotron_net, tacotron_config = load_tacotron_model(
-        "./alphabet.txt", "./tacotron.toml", "./tacotrons_ljs_24k_v1_0300000.ckpt"
-    )
+alphabet, tacotron_net, tacotron_config = load_tacotron_model(
+    "./alphabet.txt", "./tacotron.toml", "./tacotrons_ljs_24k_v1_0300000.ckpt"
+)
+
+wavegru_config, wavegru_net = load_wavegru_net(
+    "./wavegru.yaml", "./wavegru_vocoder_1024_v4_1320000.ckpt"
+)
 
-    wavegru_config, wavegru_net = load_wavegru_net(
-        "./wavegru.yaml", "./wavegru_vocoder_1024_v4_1320000.ckpt"
-    )
+wave_cpp_weight_mask = extract_weight_mask(wavegru_net)
+wavecpp = load_wavegru_cpp(
+    wave_cpp_weight_mask, wavegru_config["upsample_factors"][-1]
+)
 
-    wave_cpp_weight_mask = extract_weight_mask(wavegru_net)
-    wavecpp = load_wavegru_cpp(
-        wave_cpp_weight_mask, wavegru_config["upsample_factors"][-1]
-    )
 
+def speak(text):
     mel = text_to_mel(tacotron_net, text, alphabet, tacotron_config)
     y = mel_to_wav(wavegru_net, wavecpp, mel, wavegru_config)
     return 24_000, y