fix auto LangSegment misunderstand KO

1 year ago · e066cd93d2
parent 939971afe3
commit e066cd93d2
1 changed files with 7 additions and 3 deletions
--- a/GPT_SoVITS/inference_webui.py
+++ b/GPT_SoVITS/inference_webui.py
@ -257,9 +257,13 @@ def get_phones_and_bert(text,language):
    elif language in {"zh", "ja","auto"}:
        textlist=[]
        langlist=[]
-        LangSegment.setfilters(["zh","ja","en"])
+        LangSegment.setfilters(["zh","ja","en","ko"])
        if language == "auto":
            for tmp in LangSegment.getTexts(text):
+                if tmp["lang"] == "ko":
+                    langlist.append("zh")
+                    textlist.append(tmp["text"])
+                else:
                    langlist.append(tmp["lang"])
                    textlist.append(tmp["text"])
        else: