rbg?

gradio_demo.py

@@ -15,7 +15,7 @@ import trimesh
 from huggingface_hub import snapshot_download
 
 from gradio_model3dcolor import Model3DColor
-from gradio_model3dnormal import Model3DNormal
+# from gradio_model3dnormal import Model3DNormal
 
 code_dir = snapshot_download("sudo-ai/MeshFormer-API", token=os.environ['HF_TOKEN'])
 
@@ -92,7 +92,7 @@ def mesh_gen(tmp_dir, simplify, num_inference_steps):
     colors = (-vertex_normals + 1) / 2.0
     colors = (colors * 255).astype(np.uint8)  # Convert to 8-bit color
     print(colors.shape)
-    mesh.visual.vertex_colors = colors[..., [2, 1, 0]]  # RGB -> BGR
+    mesh.visual.vertex_colors = colors[..., [0, 2, 1]]  # RGB -> RBG
     mesh.export(f"{tmp_dir}/mesh_normal.ply", file_type="ply")
 
     color_path = ply_to_glb(f"{tmp_dir}/mesh.ply")