fix

ragg/app.py

@@ -91,18 +91,18 @@ def xtts_speak_to_file(
     if not hasattr(xtts_speak_to_file, "_model") or xtts_speak_to_file._model is None:
         import sys, builtins, torch
         from torch.serialization import add_safe_globals
-        # --- XTTS internal classes that must be trusted ---
+        # --- XTTS internal classes that must be allow-listed ---
         from TTS.tts.configs.xtts_config import XttsConfig
-        from TTS.tts.models.xtts import XttsAudioConfig
+        from TTS.tts.models.xtts import XttsAudioConfig, XttsArgs
         from TTS.config.shared_configs import BaseDatasetConfig
 
-        # Prevent interactive prompts / stdin crashes on HF
+        # Prevent interactive prompts / stdin crashes on Hugging Face
         sys.stdin = open(os.devnull)
         builtins.input = lambda *a, **kw: ""
         os.environ["COQUI_TOS_AGREED"] = "1"
 
-        # Allowlist all required XTTS classes
-        add_safe_globals([XttsConfig, XttsAudioConfig, BaseDatasetConfig])
+        # Allowlist all required XTTS classes for PyTorch 2.6+
+        add_safe_globals([XttsConfig, XttsAudioConfig, BaseDatasetConfig, XttsArgs])
 
         # Initialize the XTTS model safely
         xtts_speak_to_file._model = TTS(
@@ -113,6 +113,28 @@ def xtts_speak_to_file(
     # if not hasattr(xtts_speak_to_file, "_model") or xtts_speak_to_file._model is None:
     #     import sys, builtins, torch
     #     from torch.serialization import add_safe_globals
+    #     # --- XTTS internal classes that must be trusted ---
+    #     from TTS.tts.configs.xtts_config import XttsConfig
+    #     from TTS.tts.models.xtts import XttsAudioConfig
+    #     from TTS.config.shared_configs import BaseDatasetConfig
+
+    #     # Prevent interactive prompts / stdin crashes on HF
+    #     sys.stdin = open(os.devnull)
+    #     builtins.input = lambda *a, **kw: ""
+    #     os.environ["COQUI_TOS_AGREED"] = "1"
+
+    #     # Allowlist all required XTTS classes
+    #     add_safe_globals([XttsConfig, XttsAudioConfig, BaseDatasetConfig])
+
+    #     # Initialize the XTTS model safely
+    #     xtts_speak_to_file._model = TTS(
+    #         model_name="tts_models/multilingual/multi-dataset/xtts_v2",
+    #         gpu=False,
+    #         progress_bar=False,
+    #     )
+    # if not hasattr(xtts_speak_to_file, "_model") or xtts_speak_to_file._model is None:
+    #     import sys, builtins, torch
+    #     from torch.serialization import add_safe_globals
     #     from TTS.tts.configs.xtts_config import XttsConfig
     #     from TTS.tts.models.xtts import XttsAudioConfig