Fix cache dir for HuggingFace Spaces

app/main.py

@@ -73,12 +73,14 @@ def get_pipeline():
     global _pipeline
     if _pipeline is None:
         from app.pipeline import DiarizationPipeline
+        import os
+        cache_dir = os.getenv("CACHE_DIR", "/tmp/model_cache")
         _pipeline = DiarizationPipeline(
             device="auto",
             use_pyannote_vad=True,
             hf_token=os.getenv("HF_TOKEN"),
             max_speakers=10,
-            cache_dir="./model_cache",
+            cache_dir=cache_dir,
         )
     return _pipeline
 

models/embedder.py

@@ -22,7 +22,7 @@ class EcapaTDNNEmbedder:
     SAMPLE_RATE = 16000
     EMBEDDING_DIM = 192
 
-    def __init__(self, device: str = "auto", cache_dir: str = "./model_cache"):
+    def __init__(self, device: str = "auto", cache_dir: str = "/tmp/model_cache"):
         self.device = self._resolve_device(device)
         self.cache_dir = Path(cache_dir)
         self.cache_dir.mkdir(parents=True, exist_ok=True)