load collectivat

app.py

@@ -26,9 +26,22 @@ def carrega_bsc():
 
     return synthesizer
 
+def carrega_collectivat():
+    model_path = os.getcwd() + "/models/collectivat/fast-speech_best_model.pth"
+    config_path = os.getcwd() + "/models/collectivat/fast-speech_config.json"
+    vocoder_path = os.getcwd() + "/models/collectivat/ljspeech--hifigan_v2_model_file.pth"
+    vocoder_config_path = os.getcwd() + "/models/collectivat/ljspeech--hifigan_v2_config.json"
+    synthesizer = Synthesizer(
+        model_path, config_path, None, None, vocoder_path, vocoder_config_path
+    )
+
+    return synthesizer
+
+
 model_bsc = carrega_bsc()
 SPEAKERS = model_bsc.speakers
 
+model_collectivat = carrega_collectivat()
 
 def tts(text, speaker_idx):
     if len(text) > MAX_TXT_LEN:

models/bsc/config.json

@@ -202,7 +202,7 @@
         "use_spectral_norm_disriminator": false,
         "use_speaker_embedding": true,
         "num_speakers": 257,
-        "speakers_file": "/home/user/app/speakers.pth",
+        "speakers_file": "/home/user/app/models/bsc/speakers.pth",
         "d_vector_file": null,
         "speaker_embedding_channels": 256,
         "use_d_vector_file": false,