From 64ddece1639332a70f8157d3580a46719c6df78e Mon Sep 17 00:00:00 2001 From: litagin02 Date: Wed, 5 Jun 2024 07:14:18 +0900 Subject: [PATCH] Fix --- gradio_tabs/merge.py | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/gradio_tabs/merge.py b/gradio_tabs/merge.py index 7ee3928..3c2bd58 100644 --- a/gradio_tabs/merge.py +++ b/gradio_tabs/merge.py @@ -432,9 +432,9 @@ def simple_tts( return model.infer(text, style=style, style_weight=style_weight) -def update_two_model_names_dropdown(model_holder: TTSModelHolder): +def update_three_model_names_dropdown(model_holder: TTSModelHolder): new_names, new_files, _ = model_holder.update_model_names_for_gradio() - return new_names, new_files, new_names, new_files + return new_names, new_files, new_names, new_files, new_names, new_files def load_styles_gr(model_name_a: str, model_name_b: str): @@ -660,8 +660,15 @@ def create_merge_app(model_holder: TTSModelHolder) -> gr.Blocks: ) refresh_button.click( - lambda: update_two_model_names_dropdown(model_holder), - outputs=[model_name_a, model_path_a, model_name_b, model_path_b], + lambda: update_three_model_names_dropdown(model_holder), + outputs=[ + model_name_a, + model_path_a, + model_name_b, + model_path_b, + model_name_c, + model_path_c, + ], ) load_style_button.click(