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
+2 -2
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: