Fix HF spaces

sf3d/system.py

@@ -29,8 +29,6 @@ from sf3d.models.utils import (
 )
 from sf3d.utils import create_intrinsic_from_fov_deg, default_cond_c2w
 
-from .texture_baker import TextureBaker
-
 
 class SF3D(BaseModule):
     @dataclass
@@ -136,7 +134,6 @@ class SF3D(BaseModule):
             ),
         )
 
-        self.baker = TextureBaker()
         self.image_processor = ImageProcessor()
 
     def triplane_to_meshes(
@@ -284,6 +281,9 @@ class SF3D(BaseModule):
         bake_resolution: int,
         estimate_illumination: bool = False,
     ) -> Tuple[List[trimesh.Trimesh], dict[str, Any]]:
+        from .texture_baker import TextureBaker
+
+        baker = TextureBaker()
         batch["rgb_cond"] = self.image_processor(
             batch["rgb_cond"], self.cfg.cond_image_size
         )
@@ -314,12 +314,10 @@ class SF3D(BaseModule):
                     mesh.unwrap_uv()
 
                     # Build textures
-                    rast = self.baker.rasterize(
-                        mesh.v_tex, mesh.t_pos_idx, bake_resolution
-                    )
-                    bake_mask = self.baker.get_mask(rast)
+                    rast = baker.rasterize(mesh.v_tex, mesh.t_pos_idx, bake_resolution)
+                    bake_mask = baker.get_mask(rast)
 
-                    pos_bake = self.baker.interpolate(
+                    pos_bake = baker.interpolate(
                         mesh.v_pos,
                         rast,
                         mesh.t_pos_idx,
@@ -332,7 +330,7 @@ class SF3D(BaseModule):
                         tri_query, exclude=["density", "vertex_offset"]
                     )
 
-                    nrm = self.baker.interpolate(
+                    nrm = baker.interpolate(
                         mesh.v_nrm,
                         rast,
                         mesh.t_pos_idx,
@@ -373,7 +371,7 @@ class SF3D(BaseModule):
                             if k == "normal":
                                 # Use un-normalized tangents here so that larger smaller tris
                                 # Don't effect the tangents that much
-                                tng = self.baker.interpolate(
+                                tng = baker.interpolate(
                                     mesh.v_tng,
                                     rast,
                                     mesh.t_pos_idx,