Update app.py

app.py

@@ -1,13 +1,13 @@
-
 import os
-os.environ["COQUI_TOS_AGREED"] = "1"  # ✅ Bypass Coqui license prompt
+os.environ["COQUI_TOS_AGREED"] = "1"  # Agree to Terms of Service
 
 import torch.serialization
 
-# ✅ Fix for PyTorch 2.6 pickle loading
+# ✅ Add all necessary Coqui XTTS classes to the safe global list
 torch.serialization.add_safe_globals([
     __import__("TTS.tts.configs.xtts_config").tts.configs.xtts_config.XttsConfig,
-    __import__("TTS.tts.models.xtts").tts.models.xtts.XttsAudioConfig
+    __import__("TTS.tts.models.xtts").tts.models.xtts.XttsAudioConfig,
+    __import__("TTS.config.shared_configs").config.shared_configs.BaseDatasetConfig
 ])
 
 import gradio as gr