Update model.py

model.py

@@ -84,16 +84,18 @@ class Model:
                                                delete=False,
                                                mode='w+b')
         decode_latent_mesh(self.xm, latent).tri_mesh().write_ply(ply_path)
-        ply_path.close()  # Close the file after writing
 
-        # Open the file in binary mode for reading
-        with open(ply_path.name, 'rb') as file:
-            data = file.read()
+        mesh = trimesh.load(ply_path.name)
+        rot = trimesh.transformations.rotation_matrix(-np.pi / 2, [1, 0, 0])
+        mesh = mesh.apply_transform(rot)
+        rot = trimesh.transformations.rotation_matrix(np.pi, [0, 1, 0])
+        mesh = mesh.apply_transform(rot)
 
-        outputs = [ply_path.name, data]
-
-        return outputs
+        mesh_path = tempfile.NamedTemporaryFile(suffix='.glb', delete=False)
+        mesh.export(mesh_path.name, file_type='glb')
 
+        return mesh_path.name
+    
     def run_text(self,
                  prompt: str,
                  seed: int = 0,