add examples

app.py

@@ -1,4 +1,5 @@
 import torch
+
 torch.jit.script = lambda f: f
 
 import spaces
@@ -29,12 +30,19 @@ pipe = StableDiffusionControlNetPipeline.from_pretrained(
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 
 apply_depth = DepthDetector()
-apply_normal = NormalDetector(
-    hf_hub_download("camenduru/DSINE", filename="dsine.pt")
-)
+apply_normal = NormalDetector(hf_hub_download("camenduru/DSINE", filename="dsine.pt"))
 apply_segm = SegmDetector()
 
 
+layout_examples = [
+    [
+        "examples/layout_input.jpg",
+        "A modern bedroom",
+        "examples/layout_output.jpg",
+    ],
+]
+
+
 @spaces.GPU(duration=20)
 def generate(
     input_image,
@@ -89,13 +97,18 @@ with block:
     with gr.Row():
         gr.Markdown("## KuJiaLe Layout ControlNet Demo")
     with gr.Row():
-        input_image = gr.Image(type="numpy", label="input_image")
-    with gr.Row():
-        prompt = gr.Textbox(label="Prompt")
-    with gr.Row():
-        run_button = gr.Button(value="Run")
+        gr.Markdown(
+            "### Checkout our released model at [kujiale-ai/controlnet-layout](https://huggingface.co/kujiale-ai/controlnet-layout)"
+        )
     with gr.Row():
         with gr.Column():
+            with gr.Row():
+                input_image = gr.Image(
+                    sources="upload", type="numpy", label="Input Image", height=512
+                )
+
+            prompt = gr.Textbox(label="Prompt")
+            run_button = gr.Button(value="Run")
             with gr.Accordion("Advanced options", open=False):
                 num_samples = gr.Slider(
                     label="Images", minimum=1, maximum=2, value=1, step=1
@@ -135,9 +148,23 @@ with block:
                     value="longbody, lowres, bad anatomy, human, extra digit, fewer digits, cropped, worst quality, low quality",
                 )
 
+        with gr.Column():
+            image_gallery = gr.Gallery(
+                label="Output",
+                show_label=False,
+                elem_id="gallery",
+                height=512,
+                object_fit="contain",
+            )
     with gr.Row():
-        image_gallery = gr.Gallery(
-            label="Output", show_label=False, elem_id="gallery"
+        dummy_image_for_outputs = gr.Image(visible=False, label="Result")
+        gr.Examples(
+            fn=lambda *args: [[args[-1]], args[-2]],
+            examples=layout_examples,
+            inputs=[input_image, prompt, dummy_image_for_outputs],
+            outputs=[image_gallery, prompt],
+            run_on_click=True,
+            examples_per_page=1024,
         )
 
     ips = [
@@ -154,4 +181,4 @@ with block:
     ]
     run_button.click(fn=generate, inputs=ips, outputs=[image_gallery])
 
-block.launch(server_name='0.0.0.0')
+block.launch(server_name="0.0.0.0")