Update app.py

app.py

@@ -2,8 +2,8 @@ import os
 import torch
 import gradio as gr
 from TTS.tts.configs.xtts_config import XttsConfig
-from TTS.tts.models.xtts import XttsAudioConfig
-from TTS.config.shared_configs import BaseDatasetConfig  # ✅ Import missing class
+from TTS.tts.models.xtts import XttsAudioConfig, XttsArgs
+ # ✅ Import missing class
 from TTS.api import TTS
 
 # ✅ Accept Coqui License Automatically
@@ -11,6 +11,7 @@ os.environ["COQUI_TOS_AGREED"] = "1"
 
 # ✅ Allow missing classes for safe deserialization
 torch.serialization.add_safe_globals([XttsConfig, XttsAudioConfig, BaseDatasetConfig])
+torch.serialization.safe_globals([XttsArgs])
 
 # ✅ Force full checkpoint loading
 def safe_load_checkpoint(model_path):