Upload app.py with huggingface_hub

app.py

@@ -398,6 +398,7 @@ def prepare_multi_example() -> List[Image.Image]:
     """
     Prepare multi-image examples for the gallery.
     Adds transparent gaps between images for split_image to detect boundaries.
+    Supports variable number of views per case (up to 6).
     """
     multi_case = list(set([i.split('_')[0] for i in os.listdir("assets/example_multi_image")]))
     images = []
@@ -406,15 +407,20 @@ def prepare_multi_example() -> List[Image.Image]:
         _images = []
         # Add leading gap for boundary detection
         _images.append(np.zeros((512, gap_width, 4), dtype=np.uint8))
-        for i in range(1, 4):
-            img = Image.open(f'assets/example_multi_image/{case}_{i}.png')
+        # Load all available views (1-6)
+        for i in range(1, 7):
+            img_path = f'assets/example_multi_image/{case}_{i}.png'
+            if not os.path.exists(img_path):
+                continue
+            img = Image.open(img_path)
             W, H = img.size
             img = img.resize((int(W / H * 512), 512))
             img_array = np.array(img.convert('RGBA'))
             _images.append(img_array)
             # Add transparent gap after each image
             _images.append(np.zeros((512, gap_width, 4), dtype=np.uint8))
-        images.append(Image.fromarray(np.concatenate(_images, axis=1)))
+        if len(_images) > 1:  # Has at least one image
+            images.append(Image.fromarray(np.concatenate(_images, axis=1)))
     return images