28. Sept. 2024, 17:03

app.py

@@ -338,16 +338,12 @@ with gr.Blocks(theme=theme, css=custom_css, js=custom_js, head=custom_head, titl
                             resolution = gr.Slider(label="resolution", minimum=256, maximum=1024, value=512, step=64)
 
                 def set_inputs(tab, all_run_btns):
+                    tab = "tab_lineart" if tab is None else tab
                     # global input_options
-
-                    lineart_btn = all_run_btns[0]
-                    canny_btn = all_run_btns[1]
-
                     lineart_btn_visible = True if tab == "tab_lineart" else False
                     canny_btn_visible = True if tab == "tab_canny" else False
-                    {"visible": lineart_btn_visible, "__type__": "update"}, {"visible": canny_btn_visible, "__type__": "update"}
-
-                    return all_run_btns
+                    
+                    return {"visible": lineart_btn_visible, "__type__": "update"}, {"visible": canny_btn_visible, "__type__": "update"}
 
                 run_btn_lineart = gr.Button("Run", visible=False)
                 run_btn_canny = gr.Button("Run", visible=False)