Update app.py

app.py

@@ -16,6 +16,29 @@ from gradio.themes.utils import colors, fonts, sizes
 import rerun as rr
 from gradio_rerun import Rerun
 
+# --- Monkey Patch for cumesh to fix CUDA Error 9 ---
+# This prevents the app from crashing if fill_holes fails due to mesh density
+try:
+    import cumesh
+    # Save the original method just in case
+    if hasattr(cumesh, "Mesh"):
+        _original_fill_holes = cumesh.Mesh.fill_holes
+        
+        def safe_fill_holes(self, *args, **kwargs):
+            try:
+                return _original_fill_holes(self, *args, **kwargs)
+            except RuntimeError as e:
+                print(f"Warning: cumesh.fill_holes failed (CUDA error skipped): {e}")
+                # We simply return None or pass, effectively skipping the step
+                return None
+        
+        cumesh.Mesh.fill_holes = safe_fill_holes
+        print("Successfully monkey-patched cumesh.Mesh.fill_holes to handle CUDA errors.")
+except ImportError:
+    print("Warning: cumesh not found, skipping monkey patch.")
+except Exception as e:
+    print(f"Warning: Failed to monkey-patch cumesh: {e}")
+
 # --- Theme Configuration ---
 colors.orange_red = colors.Color(
     name="orange_red",
@@ -309,7 +332,7 @@ def generate_3d(
         
         run_id = str(uuid.uuid4())
         
-        # Robustly handle different Rerun SDK versions for recording creation
+        # Robustly handle different Rerun SDK versions
         rec = None
         if hasattr(rr, "new_recording"):
             rec = rr.new_recording(application_id="TRELLIS-3D-Viewer", recording_id=run_id)
@@ -319,8 +342,7 @@ def generate_3d(
             rr.init("TRELLIS-3D-Viewer", recording_id=run_id, spawn=False)
             rec = rr
             
-        # Log the GLB asset to Rerun
-        # We use 'path' here which is compatible with most versions and handles reading the file.
+        # Log the GLB asset using path (most robust way)
         rec.log("generated_model", rr.Asset3D(path=glb_path), static=True)
         
         # Also log the input image for context