Update app.py

app.py

@@ -22,7 +22,7 @@ from tqdm import tqdm
 from src.utils.camera_util import (FOV_to_intrinsics, get_circular_camera_poses,
                                    get_zero123plus_input_cameras)
 from src.utils.infer_util import (remove_background, resize_foreground)
-from src.utils.mesh_util import save_glb, save_obj, save_obj_with_mtl
+from src.utils.mesh_util import save_glb, save_obj
 from src.utils.train_util import instantiate_from_config
 
 zero = torch.Tensor([0]).cuda()
@@ -118,36 +118,21 @@ def make3d(images):
     mesh_basename = os.path.basename(mesh_fpath).split('.')[0]
     mesh_dirname = os.path.dirname(mesh_fpath)
     mesh_glb_fpath = os.path.join(mesh_dirname, f"{mesh_basename}.glb")
-    texmap_fpath = os.path.join(mesh_dirname, f"{mesh_basename}_tex.png")
 
     with torch.no_grad():
         planes = model.forward_planes(images, input_cameras)
-
-        # Extraer el mesh y la textura
         mesh_out = model.extract_mesh(
-            planes,
-            use_texture_map=True,  # Habilitar la generación de texturas
-            **infer_config,
-        )
-
-        # Guardar el mesh con o sin textura
-        if len(mesh_out) == 5:  # Si se genera una textura
-            vertices, faces, uvs, mesh_tex_idx, tex_map = mesh_out
-            save_obj_with_mtl(
-                vertices.data.cpu().numpy(),
-                uvs.data.cpu().numpy(),
-                faces.data.cpu().numpy(),
-                mesh_tex_idx.data.cpu().numpy(),
-                tex_map.permute(1, 2, 0).data.cpu().numpy(),
-                mesh_fpath,
-            )
-        else:
-            vertices, faces, vertex_colors = mesh_out
-            save_obj(vertices, faces, vertex_colors, mesh_fpath)
+            planes, use_texture_map=False, **infer_config)
+
+        vertices, faces, vertex_colors = mesh_out
+        vertices = vertices[:, [1, 2, 0]]
+
+        save_glb(vertices, faces, vertex_colors, mesh_glb_fpath)
+        save_obj(vertices, faces, vertex_colors, mesh_fpath)
 
         print(f"Mesh saved to {mesh_fpath}")
 
-    return mesh_fpath, mesh_glb_fpath, texmap_fpath if len(mesh_out) == 5 else None
+    return mesh_fpath, mesh_glb_fpath
 
 
 @spaces.GPU
@@ -263,12 +248,6 @@ with gr.Blocks() as demo:
                         label="(Formato GLB)",
                         interactive=False,
                     )
-                with gr.Tab("Textura"):
-                    output_texture = gr.Image(
-                        label="Textura Generada",
-                        type="pil",
-                        interactive=False
-                    )
 
     mv_images = gr.State()
 
@@ -283,7 +262,7 @@ with gr.Blocks() as demo:
     ).success(
         fn=make3d,
         inputs=[mv_images],
-        outputs=[output_model_obj, output_model_glb, output_texture]
+        outputs=[output_model_obj, output_model_glb]
     )
 
 demo.launch()