Update main.py

main.py

@@ -6,6 +6,8 @@ import gradio as gr
 from TTS.api import TTS
 from pathlib import Path
 from datetime import datetime
+from TTS.tts.configs.xtts_config import XttsConfig
+from TTS.tts.models.xtts import Xtts
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 OUTPUT_DIR = Path("outputs")
@@ -14,6 +16,10 @@ HISTORY = []
 
 tts = None
 
+
+# PyTorch 2.6 fix
+torch.serialization.add_safe_globals([XttsConfig])
+
 def get_tts():
     global tts
     if tts is None: