Update finetune_xtts_hf.py

finetune_xtts_hf.py

@@ -9,11 +9,22 @@ DATASET_PATH = "/home/user/app/dataset"        # Ruta a tu dataset
 OUTPUT_PATH = "/tmp/output_model"
 BASE_MODEL = "coqui/XTTS-v2"
 
+os.makedirs("./xtts_model", exist_ok=True)
+os.makedirs("./output", exist_ok=True)
+
+
 # === 1. Descargar modelo base de HF ===
 print("=== Descargando modelo base desde Hugging Face ===")
-subprocess.run([
-    "huggingface-cli", "download", BASE_MODEL, "--local-dir", "./xtts_model"
-], check=True)
+from huggingface_hub import snapshot_download
+
+print("=== Descargando modelo base desde Hugging Face ===")
+snapshot_download(
+    repo_id="coqui/XTTS-v2",
+    local_dir="./xtts_model",
+    local_dir_use_symlinks=False,
+    revision="main"  # aseguras que baje la última versión estable
+)
+
 
 CONFIG_PATH = "./xtts_model/config.json"
 RESTORE_PATH = "./xtts_model/model.pth"