Refactor vertex handling in generate3d function for improved performance

- Updated the vertex color assignment process to eliminate unnecessary CPU operations by directly using tensor data.
- Enhanced the creation of the mesh by utilizing tensor inputs for vertices and faces, streamlining the overall mesh generation workflow.

inference.py

@@ -84,10 +84,10 @@ def generate3d(model, rgb, ccm, device):
 
     # For each new vertex, find the nearest old vertex and copy its color
     k = 3  # or 5, etc.
-    dists, idxs = tree.query(verts.cpu().numpy(), k=k)
+    dists, idxs = tree.query(verts, k=k)
     new_colors = np.mean(color_tri[idxs], axis=1)
 
     # Create the new mesh with colors
-    mesh = trimesh.Trimesh(vertices=verts.cpu().numpy(), faces=faces.cpu().numpy(), vertex_colors=new_colors)
+    mesh = trimesh.Trimesh(vertices=verts, faces=faces, vertex_colors=new_colors)
 
     return obj_path