Fix cache directory permissions with absolute path

app.py

@@ -21,7 +21,13 @@ from pydantic import BaseModel
 
 # Seed-VC
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'seed-vc'))
-os.environ['HF_HUB_CACHE'] = os.path.join(os.path.dirname(__file__), 'checkpoints', 'hf_cache')
+
+# Hugging Face cache directory (absolute path)
+cache_dir = '/app/checkpoints'
+os.makedirs(cache_dir, exist_ok=True)
+os.environ['HF_HOME'] = cache_dir
+os.environ['HF_HUB_CACHE'] = cache_dir
+os.environ['TRANSFORMERS_CACHE'] = cache_dir
 os.environ['PYTORCH_ENABLE_MPS_FALLBACK'] = '1'
 
 # MPSを無効化してCPUを強制

seed-vc/hf_utils.py

@@ -3,10 +3,12 @@ from huggingface_hub import hf_hub_download
 
 
 def load_custom_model_from_hf(repo_id, model_filename="pytorch_model.bin", config_filename=None):
-    os.makedirs("./checkpoints", exist_ok=True)
-    model_path = hf_hub_download(repo_id=repo_id, filename=model_filename, cache_dir="./checkpoints")
+    # Use environment variable for cache directory, fallback to ./checkpoints
+    cache_dir = os.environ.get('HF_HUB_CACHE', './checkpoints')
+    os.makedirs(cache_dir, exist_ok=True)
+    model_path = hf_hub_download(repo_id=repo_id, filename=model_filename, cache_dir=cache_dir)
     if config_filename is None:
         return model_path
-    config_path = hf_hub_download(repo_id=repo_id, filename=config_filename, cache_dir="./checkpoints")
+    config_path = hf_hub_download(repo_id=repo_id, filename=config_filename, cache_dir=cache_dir)
 
     return model_path, config_path