Update app.py

app.py

@@ -79,7 +79,7 @@ tone_color_converter = ToneColorConverter(f'{ckpt_converter_en}/config.json', de
 tone_color_converter.load_ckpt(f'{ckpt_converter_en}/checkpoint.pth')
 
 
-def generate_audio(text, audio_ref, style_mode, temperature, top_P, top_K, audio_seed_input, text_seed_input, refine_text_flag, refine_text_input):
+def generate_audio(text, audio_ref, temperature, top_P, top_K, audio_seed_input, text_seed_input, refine_text_flag, refine_text_input):
     if style_mode=="default":
         source_se = torch.load(f'{ckpt_base_en}/en_default_se.pth').to(device)
         reference_speaker = audio_ref
@@ -127,10 +127,13 @@ with gr.Blocks() as demo:
 
     default_text = "Today a man knocked on my door and asked for a small donation toward the local swimming pool. I gave him a glass of water."        
     text_input = gr.Textbox(label="Input Text", lines=4, placeholder="Please Input Text...", value=default_text)
-    
+
+
     default_refine_text = "[oral_2][laugh_0][break_6]"    
     refine_text_checkbox = gr.Checkbox(label="Refine text：'oral' means add filler words, 'laugh' means add laughter, and 'break' means add a pause. (0-10) ", value=True)
     refine_text_input = gr.Textbox(label="Refine Prompt", lines=1, placeholder="Please Refine Prompt...", value=default_refine_text)
+    with gr.Column():    
+        clone_voice = gr.Audio(label="请上传您喜欢的语音文件", type="filepath")
 
     with gr.Row():
         temperature_slider = gr.Slider(minimum=0.00001, maximum=1.0, step=0.00001, value=0.3, label="Audio temperature")
@@ -157,7 +160,7 @@ with gr.Blocks() as demo:
                              outputs=text_seed_input)
         
     generate_button.click(generate_audio, 
-                          inputs=[text_input, temperature_slider, top_p_slider, top_k_slider, audio_seed_input, text_seed_input, refine_text_checkbox, refine_text_input], 
+                          inputs=[text_input, clone_voice, temperature_slider, top_p_slider, top_k_slider, audio_seed_input, text_seed_input, refine_text_checkbox, refine_text_input], 
                           outputs=audio_output)
 
 parser = argparse.ArgumentParser(description='ChatTTS demo Launch')