fix_onlyasr (#1433)

2024-08-12 10:42:52 +08:00
parent 2310bcde53
commit 62831dfcc7
2 changed files with 60 additions and 47 deletions
--- a/tools/asr/fasterwhisper_asr.py
+++ b/tools/asr/fasterwhisper_asr.py
@@ -68,10 +68,9 @@ def execute_asr(input_folder, output_folder, model_size, language, precision):

            if info.language == "zh":
                print("检测为中文文本, 转 FunASR 处理")
-                if("only_asr"not in globals()):
-                    from tools.asr.funasr_asr import \
-                        only_asr  # #如果用英文就不需要导入下载模型
-                text = only_asr(file_path)
+                if("only_asr" not in globals()):
+                    from tools.asr.funasr_asr import only_asr  #如果用英文就不需要导入下载模型
+                text = only_asr(file_path, language=info.language.lower())

            if text == '':
                for segment in segments: