Comment out KDTree functionality in generate3d function to simplify color assignment process. Updated mesh creation to set vertex colors to None, removing reliance on nearest neighbor color mapping.

inference.py

@@ -80,17 +80,17 @@ def generate3d(model, rgb, ccm, device):
     data_config['verts'] = torch.from_numpy(verts).to(device).contiguous()
     data_config['faces'] = torch.from_numpy(faces).to(device).contiguous()
 
-    # Build KDTree from original verts
-    tree = cKDTree(orig_verts)
+    # # Build KDTree from original verts
+    # tree = cKDTree(orig_verts)
 
-    # For each new vertex, find the nearest old vertex and copy its color
-    k = 3
-    dists, idxs = tree.query(verts, k=k)
-    # Use only the nearest neighbor for color assignment
-    new_colors = orig_colors[idxs[:, 0]]
+    # # For each new vertex, find the nearest old vertex and copy its color
+    # k = 3
+    # dists, idxs = tree.query(verts, k=k)
+    # # Use only the nearest neighbor for color assignment
+    # new_colors = orig_colors[idxs[:, 0]]
 
     # Create the new mesh with colors
-    mesh = trimesh.Trimesh(vertices=verts, faces=faces, vertex_colors=new_colors)
+    mesh = trimesh.Trimesh(vertices=verts, faces=faces, vertex_colors=None)
 
     # === Export OBJ/MTL/PNG ===
     obj_path = tempfile.NamedTemporaryFile(suffix=".obj", delete=False).name