Update frontend.py

Fix potential issues caused by ending with a Chinese comma
This commit is contained in:
zhuyunfeng
2024-09-29 13:29:28 +08:00
committed by GitHub
parent 8555ab4ded
commit 2c1877a5d4

View File

@@ -118,10 +118,10 @@ class CosyVoiceFrontEnd:
text = text.replace("\n", "")
text = replace_blank(text)
text = replace_corner_mark(text)
text = text.replace(".", "")
text = text.replace(".", "")
text = text.replace(" - ", "")
text = remove_bracket(text)
text = re.sub(r'[,]+$', '', text)
text = re.sub(r'[,]+$', '', text)
texts = list(split_paragraph(text, partial(self.tokenizer.encode, allowed_special=self.allowed_special), "zh", token_max_n=80,
token_min_n=60, merge_len=20, comma_split=False))
else: