diff --git a/indextts/utils/front.py b/indextts/utils/front.py index 24ddf03..ec3eb7f 100644 --- a/indextts/utils/front.py +++ b/indextts/utils/front.py @@ -68,12 +68,17 @@ class TextNormalizer: def load(self): # print(os.path.join(os.path.dirname(os.path.abspath(__file__)), "..")) # sys.path.append(model_dir) + import platform + if platform.machine() == "aarch64": + from wetext import Normalizer + self.zh_normalizer = Normalizer(remove_erhua=False,lang="zh",operator="tn") + self.en_normalizer = Normalizer(lang="en",operator="tn") + else: + from tn.chinese.normalizer import Normalizer as NormalizerZh + from tn.english.normalizer import Normalizer as NormalizerEn + self.zh_normalizer = NormalizerZh(remove_interjections=False, remove_erhua=False) + self.en_normalizer = NormalizerEn() - from tn.chinese.normalizer import Normalizer as NormalizerZh - from tn.english.normalizer import Normalizer as NormalizerEn - - self.zh_normalizer = NormalizerZh(remove_interjections=False, remove_erhua=False) - self.en_normalizer = NormalizerEn() def infer(self, text): pattern = re.compile("|".join(re.escape(p) for p in self.char_rep_map.keys())) diff --git a/requirements.txt b/requirements.txt index b813fe9..803d193 100644 --- a/requirements.txt +++ b/requirements.txt @@ -21,4 +21,5 @@ pypinyin librosa gradio tqdm -WeTextProcessing \ No newline at end of file +WeTextProcessing # arm机器如果安装失败,请注释此行 +wetext \ No newline at end of file