Fix text conversion for Chinese language

2024-08-04 16:51:17 +08:00
parent 204ccb8f3d
commit 2ecdc4e607
1 changed files with 2 additions and 1 deletions
--- a/whisper_fastapi.py
+++ b/whisper_fastapi.py
@@ -144,7 +144,8 @@ def stream_builder(
                data = segment._asdict()
                if data.get('words') is not None:
                    data["words"] = [i._asdict() for i in data["words"]]
-                data["text"] = ccc.convert(data["text"])
+                if info.language == "zh":
+                    data["text"] = ccc.convert(data["text"])
                yield data

    info_dict = info._asdict()