Update app.py

app.py

@@ -292,7 +292,11 @@ def generate_image(image_path, prompt, num_steps, guidance_scale, seed, num_imag
     gc.collect()
     torch.cuda.empty_cache()
 
-    return images
+    return gr.Gallery.update(
+        value=images,
+        columns=1 if num_images == 1 else 2,  # Show one column if one image, otherwise two
+        )
+    # return images
 
 ### Description
 title = r"""
@@ -362,7 +366,7 @@ with gr.Blocks(css=css) as demo:
                     minimum=1,
                     maximum=2,
                     step=1,
-                    value=1,
+                    value=2,
                 )
                 ip_adapter_scale = gr.Slider(
                     label="ip adapter scale",
@@ -396,7 +400,8 @@ with gr.Blocks(css=css) as demo:
                 randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
 
         with gr.Column():
-            gallery = gr.Gallery(label="Generated Images")
+            # gallery = gr.Gallery(label="Generated Images")
+            gallery = gr.Gallery(label="Generated Images").style(grid=[2], height="auto")  # Default to 2 columns
 
         submit.click(
             fn=randomize_seed_fn,
@@ -407,7 +412,8 @@ with gr.Blocks(css=css) as demo:
         ).then(
             fn=generate_image,
             inputs=[img_file, prompt, num_steps, guidance_scale, seed, num_images, ip_adapter_scale, kps_scale, canny_scale],
-            outputs=[gallery]
+            # outputs=[gallery]
+            outputs=gallery
         )
     
     gr.Markdown(Footer)