update

app.py

@@ -114,18 +114,21 @@ def infer(prompt,
     ).images[0]
     return [condi_img, image]
 
+def change_type(type):
+    return type
+
 canny_examples = [
     ["一个漂亮的女孩，高品质，超清晰，色彩鲜艳，超高分辨率，最佳品质，8k，高清，4K",
-     "image/woman_1.png", "Canny"],
+     "image/woman_1.png"],
     ["全景，一只可爱的白色小狗坐在杯子里，看向镜头，动漫风格，3d渲染，辛烷值渲染",
-    "image/dog.png", "Canny"]
+    "image/dog.png"]
 ]
 
 depth_examples = [
     ["新海诚风格，丰富的色彩，穿着绿色衬衫的女人站在田野里，唯美风景，清新明亮，斑驳的光影，最好的质量，超细节，8K画质",
-     "image/woman_2.png", "Depth"],
+     "image/woman_2.png"],
     ["一只颜色鲜艳的小鸟，高品质，超清晰，色彩鲜艳，超高分辨率，最佳品质，8k，高清，4K",
-     "image/bird.png", "Depth"]
+     "image/bird.png"]
 ]
 
 css="""
@@ -216,8 +219,6 @@ with gr.Blocks(css=css) as Kolors:
                         step=0.1,
                         value=1.0,
                     )
-            with gr.Row():
-                run_button = gr.Button("Run")
             with gr.Row():
                 canny_button = gr.Button("Canny")
                 depth_button = gr.Button("Depth")
@@ -229,7 +230,7 @@ with gr.Blocks(css=css) as Kolors:
         gr.Examples(
                 fn = infer,
                 examples = canny_examples,
-                inputs = [prompt, image, controlnet_type],
+                inputs = [prompt, image],
                 outputs = [result],
                 label = "Canny"
             )
@@ -237,7 +238,7 @@ with gr.Blocks(css=css) as Kolors:
         gr.Examples(
                 fn = infer,
                 examples = depth_examples,
-                inputs = [prompt, image, controlnet_type],
+                inputs = [prompt, image],
                 outputs = [result],
                 label = "Depth"
             )
@@ -249,14 +250,22 @@ with gr.Blocks(css=css) as Kolors:
     )
 
     canny_button.click(
+        fn = change_type,
+        input = "Canny",
+        outputs = controlnet_type
+    ).then(
         fn = infer,
-        inputs = [prompt, image, "Canny", negative_prompt, seed, randomize_seed, guidance_scale, num_inference_steps, controlnet_conditioning_scale, control_guidance_end, strength],
+        inputs = [prompt, image, controlnet_type, negative_prompt, seed, randomize_seed, guidance_scale, num_inference_steps, controlnet_conditioning_scale, control_guidance_end, strength],
         outputs = [result]
     )
 
     depth_button.click(
+        fn = change_type,
+        input = "Depth",
+        outputs = controlnet_type
+    ).then(
         fn = infer,
-        inputs = [prompt, image, "Depth", negative_prompt, seed, randomize_seed, guidance_scale, num_inference_steps, controlnet_conditioning_scale, control_guidance_end, strength],
+        inputs = [prompt, image, controlnet_type, negative_prompt, seed, randomize_seed, guidance_scale, num_inference_steps, controlnet_conditioning_scale, control_guidance_end, strength],
         outputs = [result]
     )