Update scene_weaver_core.py

scene_weaver_core.py

@@ -117,7 +117,15 @@ class SceneWeaverCore:
         logger.info(f"SceneWeaverCore initialized on {self.device}")
 
     def _setup_device(self, device: str) -> str:
-        """Setup computation device"""
+        """Setup computation device (ZeroGPU compatible)"""
+        import os
+
+        # On Hugging Face Spaces with ZeroGPU, use CPU for initialization
+        # GPU will be allocated by @spaces.GPU decorator at runtime
+        if os.getenv('SPACE_ID') is not None:
+            logger.info("Running on Hugging Face Spaces - using CPU for initialization")
+            return "cpu"
+
         if device == "auto":
             if torch.cuda.is_available():
                 return "cuda"
@@ -1174,4 +1182,4 @@ class SceneWeaverCore:
 
         status = self._inpainting_module.get_status()
         status["mode"] = self._current_mode
-        return status
+        return status