Update app.py

app.py

@@ -352,8 +352,8 @@ def generate(
     guidance_scale: float = 3,
     num_inference_steps: int = 30,
     randomize_seed: bool = False,
-    use_resolution_binning: bool = True, 
     num_images: int = 4,  # Number of images to generate
+    use_resolution_binning: bool = True, 
     progress=gr.Progress(track_tqdm=True),
 ):
     seed = int(randomize_seed_fn(seed, randomize_seed))
@@ -372,8 +372,8 @@ def generate(
     }
 
     # Use resolution binning for faster generation with less VRAM usage
-    # if use_resolution_binning:
-    #     options["use_resolution_binning"] = True
+    if use_resolution_binning:
+        options["use_resolution_binning"] = True
 
     # Generate images potentially in batches
     images = []
@@ -1491,15 +1491,15 @@ def create_ui():
             with gr.Column(scale=0.6):
                 result = gr.Gallery(
                     label="Result", 
-                    columns=2, 
-                    rows=2,
-                    show_label=False, 
-                    allow_preview=True, 
-                    object_fit="contain", 
-                    height="auto", 
-                    preview=True, 
-                    show_share_button=True, 
-                    show_download_button=True,
+                    columns=4, 
+                    # rows=2,
+                    # show_label=False, 
+                    # allow_preview=True, 
+                    # object_fit="contain", 
+                    # height="auto", 
+                    # preview=True, 
+                    # show_share_button=True, 
+                    # show_download_button=True
                 )