debug

gradio_demo.py

@@ -91,7 +91,8 @@ def mesh_gen(tmp_dir, simplify, num_inference_steps):
     vertex_normals = mesh.vertex_normals
     colors = (-vertex_normals + 1) / 2.0
     colors = (colors * 255).astype(np.uint8)  # Convert to 8-bit color
-    mesh.visual.vertex_colors = colors[..., [2, 0, 1]]  # RGB -> BRG
+    print(colors.shape)
+    mesh.visual.vertex_colors = colors[..., [2, 0, 1]] * 0  # RGB -> BRG
     mesh.export(f"{tmp_dir}/mesh_normal.ply", file_type="ply")
 
     color_path = ply_to_glb(f"{tmp_dir}/mesh.ply")