layout

app.py

@@ -236,7 +236,7 @@ def create_image(
             image=canny_map,
             controlnet_conditioning_scale=float(control_scale),
         )
-    return images
+    return images[0]
 
 
 def pil_to_cv2(image_pil):
@@ -273,7 +273,7 @@ If our work is helpful for your research or applications, please cite us via:
 If you have any questions, please feel free to open an issue or directly reach us out at <b>haofanwang.ai@gmail.com</b>.
 """
 
-block = gr.Blocks(css="footer {visibility: hidden}").queue(max_size=10, api_open=False)
+block = gr.Blocks()
 with block:
     # description
     gr.Markdown(title)
@@ -285,27 +285,26 @@ with block:
                 with gr.Row():
                     with gr.Column():
                         image_pil = gr.Image(label="Style Image", type="pil")
+                    with gr.Column():
+                        prompt = gr.Textbox(
+                            label="Prompt",
+                            value="a cat, masterpiece, best quality, high quality",
+                        )
 
-                target = gr.Radio(
-                    [
-                        "Load only style blocks",
-                        "Load style+layout block",
-                        "Load original IP-Adapter",
-                    ],
-                    value="Load only style blocks",
-                    label="Style mode",
-                )
-
-                prompt = gr.Textbox(
-                    label="Prompt",
-                    value="a cat, masterpiece, best quality, high quality",
-                )
-
-                scale = gr.Slider(
-                    minimum=0, maximum=2.0, step=0.01, value=1.0, label="Scale"
-                )
+                        scale = gr.Slider(
+                            minimum=0, maximum=2.0, step=0.01, value=1.0, label="Scale"
+                        )
 
                 with gr.Accordion(open=False, label="Advanced Options"):
+                    target = gr.Radio(
+                        [
+                            "Load only style blocks",
+                            "Load style+layout block",
+                            "Load original IP-Adapter",
+                        ],
+                        value="Load only style blocks",
+                        label="Style mode",
+                    )
                     with gr.Column():
                         src_image_pil = gr.Image(
                             label="Source Image (optional)", type="pil"
@@ -335,10 +334,10 @@ with block:
                     )
 
                     guidance_scale = gr.Slider(
-                        minimum=1,
-                        maximum=15.0,
+                        minimum=0,
+                        maximum=10.0,
                         step=0.01,
-                        value=5.0,
+                        value=0.0,
                         label="guidance scale",
                     )
                     num_inference_steps = gr.Slider(
@@ -359,26 +358,40 @@ with block:
                 generate_button = gr.Button("Generate Image")
 
             with gr.Column():
-                generated_image = gr.Gallery(label="Generated Image")
-
-        generate_button.click(
-            fn=create_image,
-            inputs=[
-                image_pil,
-                src_image_pil,
-                prompt,
-                n_prompt,
-                scale,
-                control_scale,
-                guidance_scale,
-                num_inference_steps,
-                seed,
-                target,
-                neg_content_prompt,
-                neg_content_scale,
-            ],
-            outputs=[generated_image],
-        )
+                generated_image = gr.Image(label="Generated Image")
+
+    inputs = [
+        image_pil,
+        src_image_pil,
+        prompt,
+        n_prompt,
+        scale,
+        control_scale,
+        guidance_scale,
+        num_inference_steps,
+        seed,
+        target,
+        neg_content_prompt,
+        neg_content_scale,
+    ]
+    outputs = [generated_image]
+
+    gr.on(
+        triggers=[
+            prompt.input,
+            generate_button.click,
+            guidance_scale.input,
+            scale.input,
+            control_scale.input,
+            seed.input,
+        ],
+        fn=create_image,
+        inputs=inputs,
+        outputs=outputs,
+        show_progress="minimal",
+        show_api=False,
+        trigger_mode="always_last",
+    )
 
     gr.Examples(
         examples=examples,
@@ -390,4 +403,5 @@ with block:
 
     gr.Markdown(article)
 
-block.launch()
+block.queue(api_open=False)
+block.launch(show_api=False)