directory

src/utils/download_models.py

@@ -5,17 +5,20 @@ from huggingface_hub import hf_hub_download
 
 
 def prepare_base_model():
-    local_dir = "./pretrained_weights/stable-diffusion-v1-5"
+    base_dir = os.getenv('APP_BASE_DIR', os.path.dirname(os.path.abspath(__file__)))
+    local_dir = os.path.join(base_dir, 'pretrained_weights', 'stable-diffusion-v1-5')
     os.makedirs(local_dir, exist_ok=True)
+    
     for hub_file in ["unet/config.json", "unet/diffusion_pytorch_model.bin"]:
-        path = Path(hub_file)
-        saved_path = local_dir / path
+        # Construct the full path to the file within the local_dir
+        saved_path = os.path.join(local_dir, hub_file)
         if os.path.exists(saved_path):
             continue
+        
         hf_hub_download(
             repo_id="runwayml/stable-diffusion-v1-5",
-            subfolder=PurePosixPath(path.parent),
-            filename=PurePosixPath(path.name),
+            subfolder=PurePosixPath(Path(hub_file).parent),
+            filename=PurePosixPath(Path(hub_file).name),
             local_dir=local_dir,
         )