Update app.py

app.py

@@ -27,8 +27,6 @@ from src.misc.image_io import save_interpolated_video
 from src.model.model.anysplat import AnySplat
 from src.model.ply_export import export_ply
 from src.utils.image import process_image
-import trimesh
-import numpy as np
 
 os.environ["ANYSPLAT_PROCESSED"] = f"{os.getcwd()}/proprocess_results"
 
@@ -65,7 +63,6 @@ def get_reconstructed_scene(outdir, model, device):
     )
 
     plyfile = os.path.join(outdir, "gaussians.ply")
-    glbfile = os.path.join(outdir, "gaussians.glb")
     
     export_ply(
         gaussians.means[0],
@@ -77,24 +74,9 @@ def get_reconstructed_scene(outdir, model, device):
         save_sh_dc_only=True,
     )
 
-    
-    # 1. Load PLY and preserve attributes
-    mesh = trimesh.load(plyfile, process=False)
-    
-    # 2. Check or assign vertex colors
-    if mesh.visual.vertex_colors is None or mesh.visual.vertex_colors.shape[1] < 4:
-        # Example: assume mesh.metadata['vertex_color'] holds (N×3) array
-        rgb = np.array(mesh.metadata['vertex_color'], dtype=np.uint8)
-        alpha = np.full((rgb.shape[0], 1), 255, dtype=np.uint8)
-        mesh.visual.vertex_colors = np.concatenate([rgb, alpha], axis=1)
-    
-    # 3. Export GLB
-    mesh.export(glbfile, file_type='glb')
-    print("Export complete: scene_colored.glb")
-
     # Clean up
     torch.cuda.empty_cache()
-    return glbfile, video, depth_colored
+    return plyfile, video, depth_colored
 
 
 # 2) Handle uploaded video/images --> produce target_dir + images