Update data/generate_audio.py

data/generate_audio.py

@@ -1,19 +1,20 @@
-# Fichier: /data/generate_audio.py (FIX SÉCURITÉ PYTORCH 2.6 + TACOTRON2 MAI)
+# Fichier: /data/generate_audio.py (VERSION FORCE-LOAD STABLE)
 import sys
 import os
 import json
 import wave 
-import torch # <--- IMPORTANT : On importe torch
-
-# --- FIX SÉCURITÉ PYTORCH ---
-# On autorise explicitement le composant RAdam qui bloque le chargement
-try:
-    from TTS.utils.radam import RAdam
-    if hasattr(torch.serialization, 'add_safe_globals'):
-        torch.serialization.add_safe_globals([RAdam])
-except Exception:
-    # Si l'import direct échoue, on utilise une méthode plus permissive pour PyTorch 2.6+
-    pass
+import collections
+import torch
+
+# --- LE CORRECTIF DÉFINITIF POUR PYTORCH 2.6+ ---
+# Cette fonction force PyTorch à ignorer la nouvelle restriction "weights_only"
+# qui bloque les anciens modèles Coqui TTS.
+original_load = torch.load
+def patched_load(*args, **kwargs):
+    kwargs['weights_only'] = False
+    return original_load(*args, **kwargs)
+torch.load = patched_load
+# -----------------------------------------------
 
 from TTS.api import TTS
 
@@ -25,9 +26,9 @@ text = sys.argv[1].replace('"', '')
 output_file = sys.argv[2] 
 
 try:
-    print("Tentative de chargement : tts_models/fr/mai/tacotron2-DDC", file=sys.stderr)
+    print("Tentative de chargement (Mode Forcé) : tts_models/fr/mai/tacotron2-DDC", file=sys.stderr)
     
-    # On charge le modèle
+    # Initialisation du modèle
     tts = TTS(model_name="tts_models/fr/mai/tacotron2-DDC", progress_bar=False, gpu=False)
     
     print(f"Génération en cours...", file=sys.stderr)