sunxy hai 11 meses
pai
achega
1c5c802888
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      asr_task.py

+ 1 - 1
asr_task.py

@@ -81,7 +81,7 @@ def asr_process(project, table, dt, cuda_id):
                     if video_type in ['mp4', 'm3u8']:
                     if video_type in ['mp4', 'm3u8']:
                         video_file = os.path.join(video_folder, filename)
                         video_file = os.path.join(video_folder, filename)
                         # 1. asr识别
                         # 1. asr识别
-                        asr_res_initial = get_whisper_asr(video=video_file)
+                        asr_res_initial = get_whisper_asr(audio=video_file)
                         print(video_id, asr_res_initial)
                         print(video_id, asr_res_initial)
                         # 2. 识别结果写入文件
                         # 2. 识别结果写入文件
                         asr_path = os.path.join(asr_folder, f"{video_id}.txt")
                         asr_path = os.path.join(asr_folder, f"{video_id}.txt")