Fix LangSegmenter

2 months ago · 746cb536c6
parent 74e79ae6d6
commit 746cb536c6
2 changed files with 8 additions and 0 deletions
--- a/GPT_SoVITS/TTS_infer_pack/TextPreprocessor.py
+++ b/GPT_SoVITS/TTS_infer_pack/TextPreprocessor.py
@ -159,6 +159,10 @@ class TextPreprocessor:
                        textlist.append(tmp["text"])
                else:
                    for tmp in LangSegmenter.getTexts(text):
+                        if langlist:
+                            if (tmp["lang"] == "en" and langlist[-1] == "en") or (tmp["lang"] != "en" and langlist[-1] != "en"):
+                                textlist[-1] += tmp["text"]
+                                continue
                        if tmp["lang"] == "en":
                            langlist.append(tmp["lang"])
                        else:
--- a/GPT_SoVITS/inference_webui.py
+++ b/GPT_SoVITS/inference_webui.py
@ -623,6 +623,10 @@ def get_phones_and_bert(text, language, version, final=False):
                textlist.append(tmp["text"])
        else:
            for tmp in LangSegmenter.getTexts(text):
+                if langlist:
+                    if (tmp["lang"] == "en" and langlist[-1] == "en") or (tmp["lang"] != "en" and langlist[-1] != "en"):
+                        textlist[-1] += tmp["text"]
+                        continue
                if tmp["lang"] == "en":
                    langlist.append(tmp["lang"])
                else: