fix

ragg/app.py

@@ -91,18 +91,20 @@ 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 and stdin errors on Hugging Face
+        # Prevent interactive prompts / stdin crashes on HF
         sys.stdin = open(os.devnull)
         builtins.input = lambda *a, **kw: ""
         os.environ["COQUI_TOS_AGREED"] = "1"
 
-        # Allowlist XTTS internal classes for PyTorch 2.6+
-        add_safe_globals([XttsConfig, XttsAudioConfig])
+        # Allowlist all required XTTS classes
+        add_safe_globals([XttsConfig, XttsAudioConfig, BaseDatasetConfig])
 
-        # Instantiate XTTS safely
+        # Initialize the XTTS model safely
         xtts_speak_to_file._model = TTS(
             model_name="tts_models/multilingual/multi-dataset/xtts_v2",
             gpu=False,
@@ -112,6 +114,26 @@ def xtts_speak_to_file(
     #     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
+
+    #     # Prevent interactive prompts and stdin errors on Hugging Face
+    #     sys.stdin = open(os.devnull)
+    #     builtins.input = lambda *a, **kw: ""
+    #     os.environ["COQUI_TOS_AGREED"] = "1"
+
+    #     # Allowlist XTTS internal classes for PyTorch 2.6+
+    #     add_safe_globals([XttsConfig, XttsAudioConfig])
+
+    #     # Instantiate XTTS 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
 
     #     # Prevent stdin / input errors on Hugging Face
     #     sys.stdin = open(os.devnull)