Update switch_app.py

switch_app.py

@@ -387,25 +387,30 @@ with gr.Blocks(css=css) as Kolors:
         )
 
     canny_button.click(
+        fn=switch_to_canny,
+        outputs=[canny_button]
+    ).then(
         fn=infer_canny,
         inputs=[prompt, image, ipa_image, negative_prompt, seed, randomize_seed, guidance_scale, num_inference_steps, controlnet_conditioning_scale, control_guidance_end, strength, ip_scale],
         outputs=[result, seed_used]
     )
 
     depth_button.click(
+        fn=switch_to_depth,
+        outputs=[depth_button]
+    ).then(
         fn=infer_depth,
         inputs=[prompt, image, ipa_image, negative_prompt, seed, randomize_seed, guidance_scale, num_inference_steps, controlnet_conditioning_scale, control_guidance_end, strength, ip_scale],
         outputs=[result, seed_used]
     )
 
     pose_button.click(
+        fn=switch_to_pose,
+        outputs=[pose_button]
+    ).then(
         fn=infer_pose,
         inputs=[prompt, image, ipa_image, negative_prompt, seed, randomize_seed, guidance_scale, num_inference_steps, controlnet_conditioning_scale, control_guidance_end, strength, ip_scale],
         outputs=[result, seed_used]
     )
 
-    canny_button.click(switch_to_canny, outputs=[canny_button])
-    depth_button.click(switch_to_depth, outputs=[depth_button])
-    pose_button.click(switch_to_pose, outputs=[pose_button])
-
 Kolors.queue().launch(debug=True, share=True)