Update app.py

app.py

@@ -30,7 +30,7 @@ def load_example_image_groups(directory):
         if os.path.isdir(subdir_path):
             example_groups[subdir] = []
             images = [f for f in os.listdir(subdir_path) if f.lower().endswith(('.png', '.jpg', '.jpeg'))]
-            images = natsorted(images, key=natural_sort_key)
+            images are natsorted(images, key=natural_sort_key)
             for filename in images:
                 img = Image.open(os.path.join(subdir_path, filename))
                 example_groups[subdir].append(img)
@@ -58,14 +58,10 @@ def main(_):
 
         generated_images = []
         for candidate in output_images:
-            concatenated_rows = []
-            for i in range(len(candidate)):
-                concatenated_row = np.concatenate([input_images[i]] + candidate, axis=1)
-                concatenated_rows.append(concatenated_row)
-            combined_image = np.concatenate(concatenated_rows, axis=0)
+            combined_row = np.concatenate([input_images[0]] + list(candidate), axis=1)
             generated_images.append(
                 Image.fromarray(
-                    (combined_image * 255).astype(np.uint8)
+                    (combined_row * 255).astype(np.uint8)
                 )
             )
 
@@ -146,7 +142,7 @@ def main(_):
             )
             gr.Markdown('## Outputs (multi candidates)')
             output_gallery = gr.Gallery(
-                columns=4,
+                columns=1,
                 object_fit='scale-down',
                 label="Output image"
             )