| from styletts2 import tts | |
| class style_tts: | |
| def __init__(self, model_path="epochs_2nd_00020.pth", config_path='config.yml'): | |
| self.speech_model = tts.StyleTTS2(model_checkpoint_path=model_path, config_path=config_path) | |
| def infer(self, prompt, voice, out="another_test.wav"): | |
| self.speech_model.inference(prompt, target_voice_path=voice, output_wav_file=out) | |
| return out |