Enhance generate3d function to initialize nvdiffrast context for mesh export, improving rendering capabilities. Updated export call to include rendering context and adjusted parameters for UVs and texture image, ensuring better integration with the model.

inference.py

@@ -18,8 +18,15 @@ def generate3d(model, rgb_image, xyz_image, device="cpu"):
     # CRM export expects a data dictionary
     data = {"rgb": rgb_image, "xyz": xyz_image}
 
-    # Export mesh using CRM model
-    model.export_mesh_wt_uv(data, output_dir, 0, device, 1024)
+    # Get rendering context required by xatlas and nvdiffrast
+    try:
+        import nvdiffrast.torch as dr
+        ctx = dr.RasterizeGLContext()
+    except Exception as e:
+        raise RuntimeError("Failed to initialize nvdiffrast context: " + str(e))
+
+    # Export mesh with UVs and texture image
+    model.export_mesh_wt_uv(ctx, data, base_path, 0, device, 512)
 
     # Convert .obj to .glb
     mesh = Mesh.load(obj_path, device=torch.device("cpu"))