Spaces:
Sleeping
Sleeping
File size: 772 Bytes
dcd7182 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
from IMSToucan.InferenceInterfaces.AnonFastSpeech2 import AnonFastSpeech2
class DemoTTS:
def __init__(self, model_paths, device):
self.device = device
fastspeech_path = model_paths / 'FastSpeech2_Multi' / 'trained_on_ground_truth_phonemes.pt'
hifigan_path = model_paths / 'HiFiGAN_combined' / 'best.pt'
self.model = AnonFastSpeech2(device=self.device, path_to_hifigan_model=hifigan_path,
path_to_fastspeech_model=fastspeech_path)
def read_text(self, transcription, speaker_embedding, text_is_phonemes=False):
self.model.default_utterance_embedding = speaker_embedding.to(self.device)
wav = self.model(text=transcription, text_is_phonemes=text_is_phonemes)
return wav
|