split

app.py

@@ -114,11 +114,14 @@ def infer(prompt,
     ).images[0]
     return [condi_img, image]
 
-examples = [
+canny_examples = [
     ["一个漂亮的女孩，高品质，超清晰，色彩鲜艳，超高分辨率，最佳品质，8k，高清，4K",
      "image/woman_1.png", "Canny"],
     ["全景，一只可爱的白色小狗坐在杯子里，看向镜头，动漫风格，3d渲染，辛烷值渲染",
-    "image/dog.png", "Canny"],
+    "image/dog.png", "Canny"]
+]
+
+depth_examples = [
     ["新海诚风格，丰富的色彩，穿着绿色衬衫的女人站在田野里，唯美风景，清新明亮，斑驳的光影，最好的质量，超细节，8K画质",
      "image/woman_2.png", "Depth"],
     ["一只颜色鲜艳的小鸟，高品质，超清晰，色彩鲜艳，超高分辨率，最佳品质，8k，高清，4K",
@@ -155,7 +158,8 @@ with gr.Blocks(css=css) as Kolors:
                 controlnet_type = gr.Dropdown(
                     ["Depth", "Canny"],
                     label = "Controlnet",
-                    value="Depth"
+                    value="Depth",
+                    visible=False
                 )
             with gr.Row():
                 image = gr.Image(label="Image", type="pil")
@@ -214,6 +218,9 @@ with gr.Blocks(css=css) as Kolors:
                     )
             with gr.Row():
                 run_button = gr.Button("Run")
+            with gr.Row():
+                canny_button = gr.Button("Canny")
+                depth_button = gr.Button("Depth")
             
         with gr.Column(elem_id="col-right"):
             result = gr.Gallery(label="Result", show_label=False, columns=2)
@@ -221,9 +228,18 @@ with gr.Blocks(css=css) as Kolors:
     with gr.Row():
         gr.Examples(
                 fn = infer,
-                examples = examples,
+                examples = canny_examples,
+                inputs = [prompt, image, controlnet_type],
+                outputs = [result],
+                label = "Canny"
+            )
+    with gr.Row():
+        gr.Examples(
+                fn = infer,
+                examples = depth_examples,
                 inputs = [prompt, image, controlnet_type],
-                outputs = [result]
+                outputs = [result],
+                label = "Depth"
             )
 
     run_button.click(
@@ -232,4 +248,16 @@ with gr.Blocks(css=css) as Kolors:
         outputs = [result]
     )
 
+    canny_button.click(
+        fn = infer,
+        inputs = [prompt, image, "Canny", negative_prompt, seed, randomize_seed, guidance_scale, num_inference_steps, controlnet_conditioning_scale, control_guidance_end, strength],
+        outputs = [result]
+    )
+
+    depth_button.click(
+        fn = infer,
+        inputs = [prompt, image, "Depth", negative_prompt, seed, randomize_seed, guidance_scale, num_inference_steps, controlnet_conditioning_scale, control_guidance_end, strength],
+        outputs = [result]
+    )
+
 Kolors.queue().launch(debug=True)