Update app.py

app.py

@@ -1,16 +1,29 @@
 import os
 import torch
 import gradio as gr
-from TTS.tts.configs.xtts_config import XttsConfig
-from TTS.tts.models.xtts import XttsAudioConfig, XttsArgs
- # ✅ Import missing class
+from TTS.tts.models.xtts import XttsArgs  # ✅ Import required class
 from TTS.api import TTS
 
+
 # ✅ Accept Coqui License Automatically
 os.environ["COQUI_TOS_AGREED"] = "1"
 
-# ✅ Allow missing classes for safe deserialization
-torch.serialization.add_safe_globals([XttsConfig, XttsAudioConfig])
+
+torch.serialization.safe_globals([XttsArgs])
+
+# ✅ Force full checkpoint loading
+def safe_load_checkpoint(model_path):
+    return torch.load(model_path, map_location="cpu", weights_only=False)  # ✅ Force full deserialization
+
+# ✅ Initialize TTS model
+model_name = "tts_models/multilingual/multi-dataset/xtts_v2"
+tts = TTS(model_name=model_name).to("cpu")  # ✅ Ensure CPU usage
+
+
+# ✅ Accept Coqui License Automatically
+os.environ["COQUI_TOS_AGREED"] = "1"
+
+
 torch.serialization.safe_globals([XttsArgs])
 
 # ✅ Force full checkpoint loading