Update inference.py to output glb on generate 3d

inference.py

@@ -73,27 +73,16 @@ def generate3d(model, rgb, ccm, device):
 
     start_time = time.time()
     with torch.no_grad():
+        # First export to OBJ to get the mesh with textures
+        mesh_path_obj = tempfile.NamedTemporaryFile(suffix=f"", delete=False).name
+        model.export_mesh(data_config, mesh_path_obj, tri_fea_2 = triplane_feature2)
+        
+        # Load the OBJ mesh and convert to GLB
+        mesh = Mesh.load(mesh_path_obj+".obj", bound=0.9, front_dir="+z")
         mesh_path_glb = tempfile.NamedTemporaryFile(suffix=f"", delete=False).name
-        model.export_mesh(data_config, mesh_path_glb, tri_fea_2 = triplane_feature2)
-
-        # glctx = dr.RasterizeGLContext()#dr.RasterizeCudaContext()
-        # mesh_path_obj = tempfile.NamedTemporaryFile(suffix=f"", delete=False).name
-        # model.export_mesh_wt_uv(glctx, data_config, mesh_path_obj, "", device, res=(1024,1024), tri_fea_2=triplane_feature2)
-
-        # mesh = Mesh.load(mesh_path_obj+".obj", bound=0.9, front_dir="+z")
-        # mesh_path_glb = tempfile.NamedTemporaryFile(suffix=f"", delete=False).name
-        # mesh.write(mesh_path_glb+".glb")
-
-        # # mesh_obj2 = trimesh.load(mesh_path_glb+".glb", file_type='glb')
-        # # mesh_path_obj2 = tempfile.NamedTemporaryFile(suffix=f"", delete=False).name
-        # # mesh_obj2.export(mesh_path_obj2+".obj")
-
-        # with zipfile.ZipFile(mesh_path_obj+'.zip', 'w') as myzip:
-        #     myzip.write(mesh_path_obj+'.obj', mesh_path_obj.split("/")[-1]+'.obj')
-        #     myzip.write(mesh_path_obj+'.png', mesh_path_obj.split("/")[-1]+'.png')
-        #     myzip.write(mesh_path_obj+'.mtl', mesh_path_obj.split("/")[-1]+'.mtl')
+        mesh.write(mesh_path_glb+".glb")
 
     end_time = time.time()
     elapsed_time = end_time - start_time
     print(f"uv takes {elapsed_time}s")
-    return mesh_path_glb+".obj"
+    return mesh_path_glb+".glb"