Update app.py

app.py

@@ -21,7 +21,7 @@ model_path = f'./aqua/clean_model_ckpt_steps_100000.ckpt'
 config_path = f'./aqua/config.yaml'
 spk_id = "aqua"
 
-def infer(wav_fn, tran, accelerate, auto_key):
+def infer(hubert_gpu, wav_fn, tran, accelerate, auto_key):
     model = Svc(project_name, config_path, hubert_gpu, model_path, onnx=False)
 
     if wav_fn is not None:
@@ -50,6 +50,7 @@ with app:
                     with gr.Box():
                         gr.Markdown(value="""**参数设置**""")
                         with gr.Row():
+                            use_gpu = gr.Checkbox(label="启用GPU加速（需CUDA）", value=True)
                             auto = gr.Checkbox(label="启用自动变调", value=False)
                         with gr.Row():
                             acc_vaule = gr.Slider(1, 50, value=20, interactive=True, label="加速倍率")
@@ -62,7 +63,7 @@ with app:
                     with gr.Box():
                         gr.Markdown(value="""**输出日志**""")
                         infer_msg = gr.Textbox(label="日志")
-            infer_md.click(infer, [upload_input, pitch_vaule, acc_vaule, auto], [infer_msg, out_audio])
+            infer_md.click(infer, [use_gpu, upload_input, pitch_vaule, acc_vaule, auto], [infer_msg, out_audio])
         with gr.TabItem("协议"):
             gr.Markdown(value="""
                 此为上传文件推理用GUI，需要变声功能请使用变声器\n