no message

webui.py

@@ -155,8 +155,23 @@ parser.add_argument('--model_dir',
                     default='pretrained_models/CosyVoice-300M',
                     help='local path or modelscope repo id')
 args = parser.parse_args()
-cosyvoice = CosyVoice(args.model_dir)
-sft_spk = cosyvoice.list_avaliable_spks()
+
+cosyvoice_instance = None
+
+@spaces.GPU
+def create_cosyvoice(model_dir):
+    # 在这里加入你需要的处理逻辑
+    if cosyvoice_instance is not None:
+        return cosyvoice_instance
+    
+    return CosyVoice(model_dir)
+
+@spaces.GPU
+def load_sft_options():
+    return cosyvoice.list_avaliable_spks()
+# 将原来的初始化修改为函数调用
+cosyvoice = create_cosyvoice(args.model_dir)
+
 prompt_sr, target_sr = 16000, 22050
 default_data = np.zeros(target_sr)
 
@@ -171,7 +186,9 @@ with gr.Blocks() as demo:
     with gr.Row():
         mode_checkbox_group = gr.Radio(choices=inference_mode_list, label='选择推理模式', value=inference_mode_list[0])
         instruction_text = gr.Text(label="操作步骤", value=instruct_dict[inference_mode_list[0]], scale=0.5)
-        sft_dropdown = gr.Dropdown(choices=sft_spk, label='选择预训练音色', value=sft_spk[0], scale=0.25)
+        sft_dropdown = gr.Dropdown(choices=[], label='选择预训练音色',  scale=0.25)
+        load_sft_button = gr.Button("加载预训练音色")
+        load_sft_button.click(load_sft_options, outputs=sft_dropdown)
         stream = gr.Radio(choices=stream_mode_list, label='是否流式推理', value=stream_mode_list[0][1])
         speed = gr.Number(value=1, label="速度调节(仅支持非流式推理)", minimum=0.5, maximum=2.0, step=0.1)
         with gr.Column(scale=0.25):