Update finetune_xtts_hf.py

finetune_xtts_hf.py

@@ -16,7 +16,7 @@ subprocess.run([sys.executable, "-m", "pip", "uninstall", "-y", "hf_transfer"])
 # === Configuración ===
 HF_MODEL_ID = "tu_usuario/xtts-v2-finetuned"   # <--- cambia con tu repo en HF
 HF_TOKEN = os.environ.get("HF_TOKEN")          # Debe estar definido en tu Space/entorno
-DATASET_PATH = "/tmp/dataset"        # Ruta a tu dataset
+DATASET_PATH = "/tmp/dataset/voxpopul_es_500"        # Ruta a tu dataset
 OUTPUT_PATH = "/tmp/output_model"
 BASE_MODEL = "coqui/XTTS-v2"
 
@@ -73,11 +73,10 @@ def extract_zip(zip_file_path, destination_path):
 
 # Example usage:
 zip_file = "/home/user/app/voxpopuli_es_500.zip"
-dataset_folder = "/tmp/dataset"
 
 # To protect against security vulnerabilities, it is important to sanitize the destination path.
 # This prevents an attacker from using a malicious ZIP file to write outside the destination folder.
-safe_destination = os.path.abspath(dataset_folder)
+safe_destination = os.path.abspath(DATASET_PATH)
 
 # Call the function
 extract_zip(zip_file, safe_destination)