Update app.py

app.py

@@ -8,6 +8,44 @@ import time
 import torch
 import torchaudio
 
+# ============================================================================
+# Fix for PyTorch 2.6+ - Add safe globals for TTS classes
+# PyTorch 2.6 changed the default value of `weights_only` from False to True
+# This breaks loading TTS models that use custom classes
+# ============================================================================
+import torch.serialization
+
+# First, try to import all necessary TTS classes
+try:
+    from TTS.tts.configs.xtts_config import XttsConfig
+    from TTS.tts.models.xtts import Xtts
+    from TTS.config.shared_configs import BaseDatasetConfig
+    
+    # Add all necessary TTS classes to safe globals
+    torch.serialization.add_safe_globals([
+        XttsConfig,
+        Xtts,
+        BaseDatasetConfig,
+    ])
+    print("Added TTS classes to torch safe globals")
+except Exception as e:
+    print(f"Warning: Could not add safe globals: {e}")
+
+# Fallback: Monkey patch torch.load to use weights_only=False for TTS models
+# This is needed because TTS models may contain additional custom classes
+_original_torch_load = torch.load
+
+def _patched_torch_load(*args, **kwargs):
+    # For .pth files (model checkpoints), default to weights_only=False if not explicitly set
+    if 'weights_only' not in kwargs:
+        if len(args) > 0 and isinstance(args[0], str) and args[0].endswith('.pth'):
+            kwargs['weights_only'] = False
+    return _original_torch_load(*args, **kwargs)
+
+torch.load = _patched_torch_load
+print("Applied torch.load patch for PyTorch 2.6+ compatibility")
+# ============================================================================
+
 
 #download for mecab
 os.system('python -m unidic download')
@@ -29,8 +67,6 @@ from scipy.io.wavfile import write
 from pydub import AudioSegment
 
 from TTS.api import TTS
-from TTS.tts.configs.xtts_config import XttsConfig
-from TTS.tts.models.xtts import Xtts
 from TTS.utils.generic_utils import get_user_data_dir
 
 HF_TOKEN = os.environ.get("HF_TOKEN")