Update finetune_xtts_hf.py

finetune_xtts_hf.py

@@ -14,22 +14,16 @@ os.chmod("/tmp/xtts_model", 0o777)
 os.makedirs("/tmp/output", exist_ok=True)
 os.chmod("/tmp/output", 0o777)
 
-# Cache seguro (en Spaces siempre se puede escribir en /tmp)
-hf_cache_dir = "/tmp/hf_cache"
-os.makedirs(hf_cache_dir, exist_ok=True)
-os.chmod(hf_cache_dir, 0o777)
-
-hf_cache_dir_2 = "/tmp/hf_cache/models--coqui--XTTS-v2"
-os.makedirs(hf_cache_dir_2, exist_ok=True)
-os.chmod(hf_cache_dir_2, 0o777)
-
-print("=== Descargando modelo base desde Hugging Face ===")
+# Cache seguro dentro de tu espacio
+hf_cache_dir = "/home/user/app/xtts_model_cache"
+os.makedirs(hf_cache_dir, exist_ok=True)  # Crear carpeta si no existe
 
+# Base model download
 model_dir = snapshot_download(
     repo_id="coqui/XTTS-v2",
-    local_dir="/tmp/xtts_model",   # 👈 Descargar modelo en /tmp
-    cache_dir=hf_cache_dir,        # 👈 Forzar caché también en /tmp
-    force_download=False,          # Descarga solo si no existe
+    local_dir="/tmp/xtts_model",   # modelo temporal en /tmp
+    cache_dir=hf_cache_dir,        # cache seguro dentro de tu espacio
+    force_download=False,
 )
 
 print(f"✅ Modelo descargado en: {model_dir}")