fix

ragg/app.py

@@ -92,21 +92,41 @@ 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 stdin / input errors on Hugging Face
+        # 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 the XTTS config class for torch.load()
-        add_safe_globals([XttsConfig])
+        # 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)
+    #     builtins.input = lambda *a, **kw: ""
+    #     os.environ["COQUI_TOS_AGREED"] = "1"
+
+    #     # Allowlist the XTTS config class for torch.load()
+    #     add_safe_globals([XttsConfig])
+
+    #     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
     #     import builtins