Update app.py

app.py

@@ -19,12 +19,24 @@ from tsr.system import TSR
 from tsr.utils import remove_background, resize_foreground, to_gradio_3d_orientation
 
 from torchmcubes import marching_cubes
+import sys
+import types
+import torch
 import mcubes
 
-def marching_cubes(vertices, threshold):
-    # Simula la función original usando la librería de CPU
+# --- SIMULADOR DE TORCHMCUBES PARA CPU ---
+# Esto engaña al código de TripoSR para que use PyMCubes en lugar de la versión de GPU
+def marching_cubes_proxy(vertices, threshold):
+    # Convertimos el tensor de torch a numpy para que mcubes lo procese en CPU
     v, f = mcubes.marching_cubes(vertices.detach().cpu().numpy(), threshold)
-    return torch.from_numpy(v), torch.from_numpy(f)
+    return torch.from_numpy(v.astype("float32")), torch.from_numpy(f.astype("int64"))
+
+# Creamos un módulo virtual llamado 'torchmcubes'
+mock_torchmcubes = types.ModuleType("torchmcubes")
+mock_torchmcubes.marching_cubes = marching_cubes_proxy
+sys.modules["torchmcubes"] = mock_torchmcubes
+# --------------------------------------------------------
+