Add error handling to Gallery instantiation

app.py

@@ -26,6 +26,9 @@ from trellis.representations import Gaussian, MeshExtractResult
 from trellis.utils import render_utils, postprocessing_utils
 from pathlib import Path
 import utils.hex_hura as hex_hura
+from utils.storage import (
+    upload_files_to_repo
+)
 
 import logging
 #logging.getLogger("transformers.modeling_utils").setLevel(logging.ERROR)
@@ -59,6 +62,7 @@ from utils.image_utils import (
     change_color,
     blur_image,
     open_image,
+    try_open_image,
     upscale_image,
     lerp_imagemath,
     shrink_and_paste_on_blank,
@@ -1498,7 +1502,7 @@ with gr.Blocks(css_paths="style_20250314.css", title=title, theme='Surn/beeuty',
                             )
                             with gr.Accordion("Choose Image Style*", open=True):
                                 lora_gallery = gr.Gallery(
-                                    [(open_image(image_path), title) for image_path, title in lora_models],
+                                    [(img, title) for image_path, title in lora_models if (img := try_open_image(image_path)) is not None],
                                     label="Styles",
                                     allow_preview=False, preview=False ,
                                     columns=2,

utils/image_utils.py

@@ -69,6 +69,13 @@ def get_image_from_dict(image_path):
     else:
         return image_path, False
 
+def try_open_image(image_path):
+    try:
+        return open_image(image_path)
+    except Exception as e:
+        print(f"Skipping image {image_path} due to error: {e}")
+        return None
+
 def open_image(image_path):
     """
     Opens an image from a file path or URL, or decodes a DataURL string into an image.