zhangshuai/MeloTTS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

More API features

Browse Source
This commit is contained in:
mrfakename
2024-02-26 15:57:18 -08:00
committed by GitHub
parent e17527623a
commit ed9fe64c6b
1 changed files with 11 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
melo/api.py
View File

@@ -63,16 +63,17 @@ class TTS(nn.Module):
return audio_segments return audio_segments
@staticmethod @staticmethod
def split_sentences_into_pieces(text, language): def split_sentences_into_pieces(text, language, quiet=False):
texts = split_sentence(text, language_str=language) texts = split_sentence(text, language_str=language)
if not quiet:
print(" > Text split to sentences.") print(" > Text split to sentences.")
print('\n'.join(texts)) print('\n'.join(texts))
print(" > ===========================") print(" > ===========================")
return texts return texts
def tts_to_file(self, text, speaker_id, output_path=None, sdp_ratio=0.2, noise_scale=0.6, noise_scale_w=0.8, speed=1.0): def tts_to_file(self, text, speaker_id, output_path=None, sdp_ratio=0.2, noise_scale=0.6, noise_scale_w=0.8, speed=1.0, quiet=False, format=None):
language = self.language language = self.language
texts = self.split_sentences_into_pieces(text, language) texts = self.split_sentences_into_pieces(text, language, quiet)
audio_list = [] audio_list = []
for t in texts: for t in texts:
if language in ['EN', 'ZH_MIX_EN']: if language in ['EN', 'ZH_MIX_EN']:
@@ -109,5 +110,8 @@ class TTS(nn.Module):
if output_path is None: if output_path is None:
return audio return audio
else:
if format:
soundfile.write(output_path, audio, self.hps.data.sampling_rate, format=format)
else: else:
soundfile.write(output_path, audio, self.hps.data.sampling_rate) soundfile.write(output_path, audio, self.hps.data.sampling_rate)