Update app.py

app.py

@@ -1668,7 +1668,7 @@ if __name__ == "__main__":
                     )
         print(images)
 
-        return images[0]
+        return images
     
     css = """
     #col-container{
@@ -1727,7 +1727,7 @@ if __name__ == "__main__":
                             seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=42)
                     use_progressive_upscaling = gr.Checkbox(label="use_progressive_upscaling", value=False)
             
-            output_images = gr.Image(label="Output Image", format="png")
+            output_images = gr.Gallery(label="Output Image", format="png")
             gr.Examples(
                 examples = [
                     ["Astronaut on Mars during sunset."],