Update finetune_xtts_hf.py

finetune_xtts_hf.py

@@ -2,39 +2,41 @@ import os
 import subprocess
 from huggingface_hub import HfApi, HfFolder
 
-# ======== CONFIGURACIÓN ========
-HF_TOKEN = os.environ.get("HF_TOKEN", "").strip()
-REPO_ID = os.environ.get("HF_REPO", "sob111/xttsv2-voxpopuli-es-500")
+# === Configuración ===
+HF_TOKEN = os.environ.get("HF_TOKEN")  # asegúrate de tenerlo en Secrets del Space
+HF_REPO_ID = "sob111/xttsv2-es-finetuned"  # cambia por tu repo destino
+OUTPUT_PATH = "./output_model"
 CONFIG_PATH = "./config.json"
 
-# ======== GUARDAR TOKEN ========
-if HF_TOKEN:
-    print("=== Guardando token de Hugging Face ===")
-    HfFolder.save_token(HF_TOKEN)
-else:
-    raise RuntimeError("No se encontró HF_TOKEN. Configúralo como variable de entorno.")
+# === Guardar token de Hugging Face ===
+print("=== Guardando token de Hugging Face ===")
+HfFolder.save_token(HF_TOKEN)
 
-# ======== ENTRENAMIENTO XTTSv2 ========
+# === Iniciar entrenamiento XTTSv2 ===
 print("=== Iniciando entrenamiento XTTSv2 ===")
 try:
     subprocess.run(
-        ["tts", "--config_path", CONFIG_PATH],
+        [
+            "python",
+            "TTS/bin/train_tts.py",
+            "--config_path", CONFIG_PATH
+        ],
         check=True
     )
 except subprocess.CalledProcessError:
-    raise RuntimeError("El entrenamiento XTTSv2 falló. Revisa los logs anteriores.")
+    raise RuntimeError("❌ El entrenamiento XTTSv2 falló. Revisa los logs anteriores.")
+
 print("=== Entrenamiento finalizado ===")
 
-# ======== SUBIR MODELO A HUGGING FACE ========
+# === Subir modelo fine-tune a Hugging Face ===
 print("=== Subiendo modelo fine-tune a Hugging Face ===")
 api = HfApi()
-output_dir = "./output_model"
-
+api.create_repo(repo_id=HF_REPO_ID, exist_ok=True, token=HF_TOKEN)
 api.upload_folder(
-    folder_path=output_dir,
-    repo_id=REPO_ID,
+    folder_path=OUTPUT_PATH,
+    repo_id=HF_REPO_ID,
     repo_type="model",
     token=HF_TOKEN
 )
 
-print(f"=== Modelo subido correctamente a https://huggingface.co/{REPO_ID} ===")
+print(f"✅ Fine-tuning completado y subido a {HF_REPO_ID}")