UI fix added example input

app.py

@@ -4,7 +4,10 @@ import numpy as np
 from torchvision import transforms
 from load_model import sample
 import torch
+import glob
 import random
+import os
+import pathlib
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 device = "mps" if torch.backends.mps.is_available() else device
@@ -12,6 +15,20 @@ device = "mps" if torch.backends.mps.is_available() else device
 image_size = 128
 
 
+def show_example_fn():
+    sketch = Image.open("examples/sketch.png")
+    scribble_folder = "./examples/scribbles/"
+
+    png_files = [f for f in os.listdir(scribble_folder) if f.lower().endswith(".png")]
+
+    # get random scribble
+    random_scribble = Image.open(
+        os.path.join(scribble_folder, random.choice(png_files))
+    )
+
+    return [sketch, random_scribble]
+
+
 transform = transforms.Compose(
     [
         transforms.Resize((image_size, image_size)),
@@ -60,12 +77,6 @@ with gr.Blocks(theme=theme) as demo:
 
         with gr.Column():
             output = gr.Image(type="pil", label="Output")
-            upscale_info = gr.Markdown(
-                "<p style='text-align: center; font-size: 12px;'>"
-                f"If you want to stretch the downloadable output, check the box below, the default output of neural networks is {image_size}x{image_size} "
-                "</p>"
-            )
-            upscale_button = gr.Checkbox(label="Stretch", value=False)
 
     with gr.Row():
         with gr.Column():
@@ -74,6 +85,11 @@ with gr.Blocks(theme=theme) as demo:
                 value=random.randint(0, 10000),
             )
 
+            upscale_button = gr.Checkbox(
+                label=f"Stretch (If you want to stretch the downloadable output, check the box below, the default output of neural networks is {image_size}x{image_size} )",
+                value=False,
+            )
+
         with gr.Column():
             sampling_slider = gr.Slider(
                 minimum=1,
@@ -82,9 +98,10 @@ with gr.Blocks(theme=theme) as demo:
                 label="DDPM Sampling Steps 🔄 (the higher the number of steps the higher the quality of the images)",
                 value=50,
             )
+            show_example = gr.Button(value="Show Example Input ")
 
     with gr.Row():
-        generate_button = gr.Button(value="Generate")
+        generate_button = gr.Button(value="Paint 🎨 ")
     with gr.Row():
         generate_info = gr.Markdown(
             "<p style='text-align: center; font-size: 16px;'>"
@@ -92,6 +109,14 @@ with gr.Blocks(theme=theme) as demo:
             "</p>"
         )
 
+    show_example.click(
+        show_example_fn,
+        inputs=[],
+        outputs=[sketch_input, scribbles_input],
+        concurrency_limit=1,
+        trigger_mode="once",
+    )
+
     generate_button.click(
         process_images,
         inputs=[
@@ -107,4 +132,4 @@ with gr.Blocks(theme=theme) as demo:
     )
 
 if __name__ == "__main__":
-    demo.queue().launch(max_threads=1)
+    demo.launch()