Update app.py

app.py

@@ -50,7 +50,7 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
 ).to(device)
 pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
 
-# <compile speedup >
+# Compile speedup
 if USE_TORCH_COMPILE:
     pipe.compile()
 
@@ -103,7 +103,7 @@ def generate(
     if use_resolution_binning:
         options["use_resolution_binning"] = True
 
-    # Images potential batches
+    # Generate images in batches
     images = []
     for i in range(0, num_images, BATCH_SIZE):
         batch_options = options.copy()
@@ -128,7 +128,8 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
                 )
                 run_button = gr.Button("Run", scale=0, variant="primary")
 
-            result = gr.Image(show_label=False, format="png")
+            # Changed from gr.Image to gr.Gallery to handle multiple images
+            result = gr.Gallery(label="Result", columns=2, show_label=False)
             
     with gr.Accordion("Advanced options", open=False, visible=True):
         num_images = gr.Slider(