transkun

app.py

@@ -325,14 +325,14 @@ def process_audio(input_file, use_cuda=True, use_quantize=True, progress=gr.Prog
         quantized_output_file = None
 
         # 设置设备参数
-        device_param = "--cuda" if use_cuda and cuda_available else ""
+        device_param = "cuda" if use_cuda and cuda_available else "cpu"
 
         start_time = time.time()
         progress(file_progress_offset, desc="准备转录...")
 
         # 使用命令行调用transkun
         progress(file_progress_offset + 0.3 * file_progress_scale, desc="转录中...")
-        cmd = ["transkun", input_file, str(output_file), device_param]
+        cmd = ["transkun", input_file, str(output_file), "--device", device_param]
 
         # 执行命令
         process = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)