examples

.gitignore

@@ -1,3 +1,4 @@
 **/midas_models/
 stable-diffusion-2-depth/
-__pycache__
+__pycache__
+gradio_cached_examples

README.md

@@ -1,6 +1,6 @@
 ---
 title: Stablediffusion Depth2img
-emoji: 💩
+emoji: 🧊🖼
 colorFrom: green
 colorTo: yellow
 sdk: gradio

stablediffusion/scripts/gradio/depth2img.py

@@ -158,7 +158,7 @@ sampler = initialize_model(sys.argv[1], sys.argv[2])
 block = gr.Blocks().queue()
 with block:
     with gr.Row():
-        gr.Markdown("## Stable Diffusion Depth2Img")
+        gr.Markdown("## Stable Diffusion 2 Depth2Img")
 
     with gr.Row():
         with gr.Column():
@@ -187,9 +187,23 @@ with block:
         with gr.Column():
             gallery = gr.Gallery(label="Generated images", show_label=False).style(
                 grid=[2], height="auto")
-
+    gr.Examples(
+        examples=[
+            ["./examples/baby.jpg",
+             "high definition photo of a baby astronaut space walking at the international space station with earth seeing from above in the background",
+             50, 1, 9.0, 123123123, 0.0, 0.8],
+            ["./examples/gol.jpg",
+             "professional photo of a Elmo jumping between two high rises, beautiful colorful city landscape in the background",
+             50, 1, 9.0, 1734133747, 0.0, 0.9]
+        ],
+        inputs=[input_image, prompt, ddim_steps,
+                num_samples, scale, seed, eta, strength],
+        outputs=[gallery],
+        fn=predict,
+        cache_examples=True,
+    )
     run_button.click(fn=predict, inputs=[
                      input_image, prompt, ddim_steps, num_samples, scale, seed, eta, strength], outputs=[gallery])
 
 
-block.launch()
+block.launch(show_api=False)