Update cleaners.py

2023-04-20 15:19:52 +08:00
parent b3e7ad0e50
commit 8e1893daf7
1 changed files with 6 additions and 11 deletions
@@ -29,19 +29,14 @@ def korean_cleaners(text):
    return text


-# def chinese_cleaners(text):
-#     '''Pipeline for Chinese text'''
-#     text = number_to_chinese(text)
-#     text = chinese_to_bopomofo(text)
-#     text = latin_to_bopomofo(text)
-#     text = re.sub(r'([ˉˊˇˋ˙])$', r'\1。', text)
-#     return text
-
 def chinese_cleaners(text):
-    from pypinyin import Style, pinyin
+    '''Pipeline for Chinese text'''
    text = text.replace("[ZH]", "")
-    phones = [phone[0] for phone in pinyin(text, style=Style.TONE3)]
-    return ' '.join(phones)
+    text = number_to_chinese(text)
+    text = chinese_to_bopomofo(text)
+    text = latin_to_bopomofo(text)
+    text = re.sub(r'([ˉˊˇˋ˙])$', r'\1。', text)
+    return text


 def zh_ja_mixture_cleaners(text):