fix(core): split error on single sentence

2noise · Jan 13, 2025 · b5e452e · b5e452e
1 parent 25cf2bc
commit b5e452e
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/ChatTTS/core.py b/ChatTTS/core.py
@@ -225,12 +225,15 @@ def infer(
             if "\n" in text:
                 text = text.split("\n")
             else:
-                text = re.split(r"(?<=[。(.\s)])", text)
+                text = re.split(r"(?<=。)|(?<=\.\s)", text)
                 nt = []
-                for t in text:
-                    if t:
-                        nt.append(t)
-                text = nt
+                if isinstance(text, list):
+                    for t in text:
+                        if t:
+                            nt.append(t)
+                    text = nt
+                else:
+                    text = [text]
             self.logger.info("split text into %d parts", len(text))
             self.logger.debug("%s", str(text))