28. Sept. 2024, 16:39

app.py

@@ -288,7 +288,7 @@ with gr.Blocks(theme=theme, css=custom_css, js=custom_js, head=custom_head, titl
                     if tab_name == "Lineart":
                         input_options = [preprocessor_name, invert]
                     return {"preprocessor_name": preprocessor_name, "invert": invert, "resolution": resolution}
-                tab_lineart.select(fn=set_inputs(), inputs=tab_lineart, outputs=None)
+                
                 ## TAB Canny
                 with gr.Tab("Canny Edge") as tab_canny:
                     with gr.Row():
@@ -298,8 +298,15 @@ with gr.Blocks(theme=theme, css=custom_css, js=custom_js, head=custom_head, titl
                             low_threshold = gr.Slider(label="low_threshold", minimum=1, maximum=255, value=100, step=1)
                             high_threshold = gr.Slider(label="high_threshold", minimum=1, maximum=255, value=200, step=1)
                             resolution = gr.Slider(label="resolution", minimum=256, maximum=1024, value=512, step=64)
-                            run_btn_canny = gr.Button("Run")
 
+
+                run_btn_lineart = gr.Button("Run", visible=False)
+                run_btn_canny = gr.Button("Run", visible=False)
+                
+                all_run_btns = [run_btn_lineart, run_btn_canny]
+                
+                tab_lineart.select(fn=set_inputs("Lineart"), inputs=None, outputs=all_run_btns)
+                
             with gr.Column(scale=2):
                 gallery = gr.Gallery(label="Generated images", show_label=False, interactive=False, format="png", elem_id="output_gallery", elem_classes="output-gallery", columns=[3], rows=[2], object_fit="contain", height="auto", type="filepath")