Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix bug where sometimes the second sentence disappears if it doesn't end with punctuation

This commit is contained in:
Reuben Morais 2020-02-13 16:03:30 +01:00
parent b539ffafc0
commit 995eb1bf07
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
server/synthesizer.py
View File

@ -122,7 +122,7 @@ class Synthesizer(object):
self.ap.save_wav(wav, path) self.ap.save_wav(wav, path)
def split_into_sentences(self, text): def split_into_sentences(self, text):
text = " " + text + " " text = " " + text + " <stop>"
text = text.replace("\n", " ") text = text.replace("\n", " ")
text = re.sub(prefixes, "\\1<prd>", text) text = re.sub(prefixes, "\\1<prd>", text)
text = re.sub(websites, "<prd>\\1", text) text = re.sub(websites, "<prd>\\1", text)
@ -149,15 +149,13 @@ class Synthesizer(object):
text = text.replace("<prd>", ".") text = text.replace("<prd>", ".")
sentences = text.split("<stop>") sentences = text.split("<stop>")
sentences = sentences[:-1] sentences = sentences[:-1]
sentences = [s.strip() for s in sentences] sentences = list(filter(None, [s.strip() for s in sentences])) # remove empty sentences
return sentences return sentences
def tts(self, text): def tts(self, text):
wavs = [] wavs = []
sens = self.split_into_sentences(text) sens = self.split_into_sentences(text)
print(sens) print(sens)
if not sens:
sens = [text+'.']
for sen in sens: for sen in sens:
# preprocess the given text # preprocess the given text
inputs = text_to_seqvec(sen, self.tts_config, self.use_cuda) inputs = text_to_seqvec(sen, self.tts_config, self.use_cuda)