Refactor Synthesizer initialization parameters

Author: wannaphong

pythaitts/pretrained/khanomtan_tts.py

@@ -47,28 +47,22 @@ def load_synthesizer(self, mode):
         if mode=="best_model":
             self.best_model_path = hf_hub_download(repo_id="wannaphong/khanomtan-tts-v{0}".format(self.version),filename=self.best_model_path_name,force_filename="best_model-v{0}.pth".format(self.version))
             self.synthesizer = Synthesizer(
-                self.best_model_path,
-                self.config_path,
-                self.speakers_path,
-                self.languages_path,
-                None,
-                None,
-                self.speaker_encoder_model_path,
-                self.speaker_encoder_config_path,
-                False
+                tts_checkpoint=self.best_model_path,
+                tts_config_path=self.config_path,
+                tts_speakers_file=self.speakers_path,
+                tts_languages_file=self.languages_path,
+                encoder_checkpoint=self.speaker_encoder_model_path,
+                encoder_config=self.speaker_encoder_config_path
             )
         else:
             self.last_checkpoint_model_path = hf_hub_download(repo_id="wannaphong/khanomtan-tts-v{0}".format(self.version),filename=self.last_checkpoint_model_path_name,force_filename="last_checkpoint-v{0}.pth".format(self.version))
             self.synthesizer = Synthesizer(
-                self.last_checkpoint_model_path,
-                self.config_path,
-                self.speakers_path,
-                self.languages_path,
-                None,
-                None,
-                self.speaker_encoder_model_path,
-                self.speaker_encoder_config_path,
-                False
+                tts_checkpoint=self.last_checkpoint_model_path,
+                tts_config_path=self.config_path,
+                tts_speakers_file=self.speakers_path,
+                tts_languages_file=self.languages_path,
+                encoder_checkpoint=self.speaker_encoder_model_path,
+                encoder_config=self.speaker_encoder_config_path
             )
     def __call__(self, text: str, speaker_idx: str, language_idx: str, return_type: str = "file", filename: str = None):
         wavs = self.synthesizer.tts(text, speaker_idx, language_idx)
@@ -80,4 +74,4 @@ def __call__(self, text: str, speaker_idx: str, language_idx: str, return_type:
         else:
             with tempfile.NamedTemporaryFile(suffix = ".wav", delete = False) as fp:
                 self.synthesizer.save_wav(wavs, fp)
-        return fp.name
+        return fp.name