Update app.py

app.py

@@ -185,7 +185,6 @@ def generate_images():
         "--seeds=3-5,7,9,12-14,16-26,29,31,32,34,40,41",
         f"--network={MODEL_PATH}"
     ]
-    
     try:
         subprocess.run(command, check=True, capture_output=True, text=True)
     except subprocess.CalledProcessError as e:
@@ -223,27 +222,24 @@ with gr.Blocks() as demo:
     file_paths = []
 
     with gr.Row():
-        for row in range(2):  # 2 rows
-            with gr.Column():
-                for col in range(5):  # 5 columns
-                    index = row * 5 + col
-                    with gr.Column():
-                        img = gr.Image(width=200, height=200)
-                        image_components.append(img)
+        for i in range(10):  # 10 images
+            with gr.Column(scale=1):
+                img = gr.Image(width=200, height=200)
+                image_components.append(img)
 
-                        file_path = gr.Textbox(visible=False)
-                        file_paths.append(file_path)
+                file_path = gr.Textbox(visible=False)
+                file_paths.append(file_path)
 
-                        save_btn = gr.Button("💾 Save")
-                        output = gr.Textbox(visible=False)
-                        outputs.append(output)
+                save_btn = gr.Button("💾 Save")
+                output = gr.Textbox(visible=False)
+                outputs.append(output)
 
-                        save_btn.click(fn=save_image, inputs=file_path, outputs=output)
-                        save_buttons.append(save_btn)
+                save_btn.click(fn=save_image, inputs=file_path, outputs=output)
+                save_buttons.append(save_btn)
 
     def generate_and_display_images():
         image_paths = get_random_images()
-        return image_paths + image_paths  # for setting file_paths and images
+        return image_paths + image_paths  # one set for images, one for hidden file paths
 
     generate_button.click(
         fn=generate_and_display_images,
@@ -252,4 +248,3 @@ with gr.Blocks() as demo:
 
 if __name__ == "__main__":
     demo.launch()
-