Update app.py

app.py

@@ -113,7 +113,7 @@ with gr.Blocks() as demo:
                         minimum=2,
                         maximum=100,
                         step=1,
-                        value=8,
+                        value=16,
                         label="Number of Inference Steps",
                     )
 
@@ -128,7 +128,7 @@ with gr.Blocks() as demo:
                     generate_images_button_denoise = gr.Button("Generate Images")
 
                 with gr.Column():
-                    images_output_denoise = gr.Gallery(label="Images", selected_index=0, rows=1)
+                    images_output_denoise = gr.Gallery(label="Images", selected_index=0, height=512)
                     gif_denoise = gr.Image(label="GIF")
                     zip_output_denoise = gr.File(label="Download ZIP")
 
@@ -178,13 +178,13 @@ with gr.Blocks() as demo:
                         minimum=2,
                         maximum=100,
                         step=1,
-                        value=8,
+                        value=16,
                         label="Number of Inference Steps per Image",
                     )
                     generate_images_button_seed = gr.Button("Generate Images")
 
                 with gr.Column():
-                    images_output_seed = gr.Gallery(label="Images", selected_index=0, rows=1)
+                    images_output_seed = gr.Gallery(label="Images", selected_index=0, height=512)
                     zip_output_seed = gr.File(label="Download ZIP")
 
         generate_images_button_seed.click(
@@ -222,7 +222,7 @@ with gr.Blocks() as demo:
                         minimum=2,
                         maximum=100,
                         step=1,
-                        value=8,
+                        value=16,
                         label="Number of Inference Steps per Image",
                     )
 
@@ -237,7 +237,7 @@ with gr.Blocks() as demo:
                     generate_images_button_perturb = gr.Button("Generate Images")
 
                 with gr.Column():
-                    images_output_perturb = gr.Gallery(label="Image", selected_index=0, rows=1)
+                    images_output_perturb = gr.Gallery(label="Image", selected_index=0, height=512)
                     zip_output_perturb = gr.File(label="Download ZIP")
 
         generate_images_button_perturb.click(
@@ -293,7 +293,7 @@ with gr.Blocks() as demo:
                         minimum=2,
                         maximum=100,
                         step=1,
-                        value=8,
+                        value=16,
                         label="Number of Inference Steps per Image",
                     )
 
@@ -308,7 +308,7 @@ with gr.Blocks() as demo:
                     generate_images_button_circular = gr.Button("Generate Images")
 
                 with gr.Column():
-                    images_output_circular = gr.Gallery(label="Image", selected_index=0, rows=1)
+                    images_output_circular = gr.Gallery(label="Image", selected_index=0, height=512)
                     gif_circular = gr.Image(label="GIF")
                     zip_output_circular = gr.File(label="Download ZIP")
 
@@ -352,7 +352,7 @@ with gr.Blocks() as demo:
                         minimum=2,
                         maximum=100,
                         step=1,
-                        value=8,
+                        value=16,
                         label="Number of Inference Steps per Image",
                     )
 
@@ -502,7 +502,7 @@ with gr.Blocks() as demo:
                         minimum=2,
                         maximum=100,
                         step=1,
-                        value=8,
+                        value=16,
                         label="Number of Inference Steps per Image",
                     )
                     guidance_scale_values = gr.Textbox(
@@ -522,7 +522,7 @@ with gr.Blocks() as demo:
                 with gr.Column():
                     images_output_guidance = gr.Gallery(
                         label="Images", selected_index=0,
-                        rows=1,
+                        height=512,
                     )
                     zip_output_guidance = gr.File(label="Download ZIP")
 
@@ -540,53 +540,53 @@ with gr.Blocks() as demo:
             fn=generate_seed_vis, inputs=[seed_guidance], outputs=[seed_vis_guidance]
         )
 
-        # with gr.TabItem("Inpainting"):
-        #     gr.Markdown("Inpaint the image based on the prompt.")
-        #     gr.HTML(read_html("html/inpainting.html"))
-
-        #     with gr.Row():
-        #         with gr.Column():
-        #             uploaded_img_inpaint = gr.Image(
-        #                 source="upload", tool="sketch", type="pil", label="Upload"
-        #             )
-        #             prompt_inpaint = gr.Textbox(
-        #                 lines=1, label="Prompt", value="sunglasses"
-        #             )
-        #             num_inference_steps_inpaint = gr.Slider(
-        #                 minimum=2,
-        #                 maximum=100,
-        #                 step=1,
-        #                 value=8,
-        #                 label="Number of Inference Steps per Image",
-        #             )
-
-        #             with gr.Row():
-        #                 seed_inpaint = gr.Slider(
-        #                     minimum=0, maximum=100, step=1, value=14, label="Seed"
-        #                 )
-        #                 seed_vis_inpaint = gr.Plot(
-        #                     value=generate_seed_vis(14), label="Seed"
-        #                 )
-
-        #             inpaint_button = gr.Button("Inpaint")
-
-        #         with gr.Column():
-        #             images_output_inpaint = gr.Image(label="Output")
-        #             zip_output_inpaint = gr.File(label="Download ZIP")
-
-        # inpaint_button.click(
-        #     fn=inpaint,
-        #     inputs=[
-        #         uploaded_img_inpaint,
-        #         num_inference_steps_inpaint,
-        #         seed_inpaint,
-        #         prompt_inpaint,
-        #     ],
-        #     outputs=[images_output_inpaint, zip_output_inpaint],
-        # )
-        # seed_inpaint.change(
-        #     fn=generate_seed_vis, inputs=[seed_inpaint], outputs=[seed_vis_inpaint]
-        # )
+        with gr.TabItem("Inpainting"):
+            gr.Markdown("Inpaint the image based on the prompt.")
+            gr.HTML(read_html("html/inpainting.html"))
+    
+            with gr.Row():
+                with gr.Column():
+                    uploaded_img_inpaint = gr.Sketchpad(
+                        sources="upload", brush=gr.Brush(colors=["#ffff00"]), type="pil", label="Upload"
+                    )
+                    prompt_inpaint = gr.Textbox(
+                        lines=1, label="Prompt", value="sunglasses"
+                    )
+                    num_inference_steps_inpaint = gr.Slider(
+                        minimum=2,
+                        maximum=100,
+                        step=1,
+                        value=16,
+                        label="Number of Inference Steps per Image",
+                    )
+
+                    with gr.Row():
+                        seed_inpaint = gr.Slider(
+                            minimum=0, maximum=100, step=1, value=14, label="Seed"
+                        )
+                        seed_vis_inpaint = gr.Plot(
+                            value=generate_seed_vis(14), label="Seed"
+                        )
+
+                    inpaint_button = gr.Button("Inpaint")
+
+                with gr.Column():
+                    images_output_inpaint = gr.Image(label="Output")
+                    zip_output_inpaint = gr.File(label="Download ZIP")
+
+        inpaint_button.click(
+            fn=inpaint,
+            inputs=[
+                uploaded_img_inpaint,
+                num_inference_steps_inpaint,
+                seed_inpaint,
+                prompt_inpaint,
+            ],
+            outputs=[images_output_inpaint, zip_output_inpaint],
+        )
+        seed_inpaint.change(
+            fn=generate_seed_vis, inputs=[seed_inpaint], outputs=[seed_vis_inpaint]
+        )
 
     with gr.Tab("CLIP Space"):
 
@@ -899,7 +899,7 @@ with gr.Blocks() as demo:
                         minimum=2,
                         maximum=100,
                         step=1,
-                        value=8,
+                        value=16,
                         label="Number of Inference Steps per Image",
                     )
 
@@ -916,7 +916,7 @@ with gr.Blocks() as demo:
                 with gr.Column():
                     images_output_interpolate = gr.Gallery(
                         label="Interpolated Images", selected_index=0,
-                        rows=1,
+                        height=512,
                     )
                     gif_interpolate = gr.Image(label="GIF")
                     zip_output_interpolate = gr.File(label="Download ZIP")
@@ -960,7 +960,7 @@ with gr.Blocks() as demo:
                         minimum=2,
                         maximum=100,
                         step=1,
-                        value=8,
+                        value=16,
                         label="Number of Inference Steps per Image",
                     )