Update src/generate_utils/generate.py

src/generate_utils/generate.py

@@ -26,8 +26,8 @@ class PressureGenerator:
     }
 
     def __init__(self, 
-                 ckpt_dir="/app/src/generate_utils/lib/ckpt/pressurepose_20251222_180032",
-                 smpl_model_dir="/app/src/smpl_models",
+                 ckpt_dir="src/generate_utils/lib/ckpt/pressurepose_20251222_180032",
+                 smpl_model_dir="src/smpl_models",
                  device="cpu"):
         """
         初始化生成器：加载配置、权重和 SMPL 模型。
@@ -58,7 +58,13 @@ class PressureGenerator:
     def _load_config(self):
         config_path = os.path.join(self.ckpt_dir, 'config.yaml')
         if not os.path.exists(config_path):
-            raise FileNotFoundError(f"Config not found at {config_path}")
+            from huggingface_hub import snapshot_download
+            snapshot_download(
+                "yolozyk/PaGe",
+                local_dir="src/generate_utils/lib/ckpt/",
+                local_dir_use_symlinks=False,
+                ignore_patterns=["*.safetensors", ".gitattributes"],
+            )
         with open(config_path, 'r') as f:
             return yaml.safe_load(f)
 
@@ -70,7 +76,7 @@ class PressureGenerator:
             from huggingface_hub import snapshot_download
             snapshot_download(
                 "yolozyk/PaGe",
-                local_dir="/app/src/generate_utils/lib/ckpt/",
+                local_dir="src/generate_utils/lib/ckpt/",
                 local_dir_use_symlinks=False,
                 ignore_patterns=["*.safetensors", ".gitattributes"],
             )