Update app.py

app.py

@@ -1,14 +1,16 @@
 import gradio as gr
 import os
-import torch  # ✅ Import torch first
+import torch
+from TTS.tts.models.xtts import XttsAudioConfig  # ✅ Import the correct class
 from TTS.tts.configs.xtts_config import XttsConfig
 from TTS.api import TTS
 
-
-
 # ✅ Accept Coqui License Automatically
 os.environ["COQUI_TOS_AGREED"] = "1"
-torch.serialization.add_safe_globals([XttsConfig])
+
+# ✅ Fix PyTorch deserialization issue
+torch.serialization.add_safe_globals([XttsAudioConfig])  # ✅ This fixes the issue
+
 
 # Load the voice cloning model
 tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2").to("cpu")