diff --git a/src/blackbox/melotts.py b/src/blackbox/melotts.py index 6be8c5f..9ae3d9d 100644 --- a/src/blackbox/melotts.py +++ b/src/blackbox/melotts.py @@ -26,9 +26,9 @@ class MeloTTS(Blackbox): device: str language: str speaker: str - - @inject - def __init__(self, melo_config: MeloConf) -> None: + + @logging_time(logger=logger) + def model_init(self, melo_config: MeloConf) -> None: self.speed = melo_config.speed self.device = melo_config.device self.language = melo_config.language @@ -45,6 +45,10 @@ class MeloTTS(Blackbox): self.url = melo_config.url logging.info('#### Initializing MeloTTS Service in ' + self.device + ' mode...') + @inject + def __init__(self, melo_config: MeloConf) -> None: + self.model_init(melo_config) + def __call__(self, *args, **kwargs): return self.processing(*args, **kwargs)