UI enhancements

gradio_demo/app.py

@@ -189,15 +189,12 @@ with gr.Blocks() as demo:
     """)
     with gr.Row():
         with gr.Column():
-            lq_img = gr.Image(label="Low-quality image", type="pil")
-            gr.Examples(
-                examples = ["assets/lady.png", "assets/man.png", "assets/dog.png", "assets/panda.png", "assets/sculpture.png", "assets/cottage.png", "assets/Naruto.png", "assets/Konan.png"],
-                inputs = [lq_img]
-            )
-        with gr.Column():
+            lq_img = gr.Image(label="Low-quality image", type="pil")      
+            
             with gr.Row():
                 steps = gr.Number(label="Steps", value=30, step=1)
                 cfg_scale = gr.Number(label="CFG Scale", value=7.0, step=0.1)
+            
             with gr.Row():
                 height = gr.Number(label="Height", value=1024, step=1)
                 weight = gr.Number(label="Weight", value=1024, step=1)
@@ -207,14 +204,19 @@ with gr.Blocks() as demo:
             preview_start = gr.Slider(label="Preview Start", value=0, minimum=0, maximum=30, step=1)
             prompt = gr.Textbox(label="Restoration prompts (Optional)", placeholder="")
             mode = gr.Checkbox(label="Creative Restoration", value=False)
-    with gr.Row():
-        with gr.Row():
-            restore_btn = gr.Button("InstantIR magic!")
-            clear_btn = gr.ClearButton()
-        index = gr.Slider(label="Restoration Previews", value=29, minimum=0, maximum=29, step=1)
-    with gr.Row():
-        output = gr.Image(label="InstantIR restored", type="pil")
-        preview = gr.Image(label="Preview", type="pil")
+    
+            with gr.Row():
+                restore_btn = gr.Button("InstantIR magic!")
+                clear_btn = gr.ClearButton()
+            gr.Examples(
+                    examples = ["assets/lady.png", "assets/man.png", "assets/dog.png", "assets/panda.png", "assets/sculpture.png", "assets/cottage.png", "assets/Naruto.png", "assets/Konan.png"],
+                    inputs = [lq_img]
+                )
+        with gr.Column():
+            output = gr.Image(label="InstantIR restored", type="pil")
+            index = gr.Slider(label="Restoration Previews", value=29, minimum=0, maximum=29, step=1)
+            preview = gr.Image(label="Preview", type="pil")
+       
     pipe_out = gr.Gallery(visible=False)
     clear_btn.add([lq_img, output, preview])
     restore_btn.click(
@@ -238,11 +240,6 @@ with gr.Blocks() as demo:
     1. Check the `Creative Restoration` checkbox;
     2. Input your text prompts in the `Restoration prompts` textbox;
     3. Set `Start Free Rendering` slider to a medium value (around half of the `steps`) to provide adequate room for InstantIR creation.
-    
-    ## Examples
-    Here are some examplar usage of InstantIR:
-    """)
-    # examples = gr.Gallery(label="Examples")
 
     gr.Markdown(
     """