Update app.py

app.py

@@ -1,23 +1,21 @@
 import os
 import torch
 import gradio as gr
+from TTS.tts.configs.xtts_config import XttsConfig  # ✅ Required for deserialization
 from TTS.api import TTS
 
 # ✅ Accept Coqui License Automatically
 os.environ["COQUI_TOS_AGREED"] = "1"
 
-# ✅ Ensure CPU-only execution (disable GPU)
-device = "cpu"
-
-# ✅ Manually Load XTTS Model (Without `XttsConfig`)
-model_name = "tts_models/multilingual/multi-dataset/xtts_v2"
-model_path = f"{os.getenv('HOME')}/.local/share/tts/{model_name.replace('/', '--')}"  # Custom model path
-
-tts = TTS(model_name=model_name).to(device)
-
-
+# ✅ Allow `XttsConfig` in PyTorch's safe globals
+torch.serialization.add_safe_globals([XttsConfig])
 
+# ✅ Force full checkpoint deserialization
+def safe_load_checkpoint(model_path):
+    return torch.load(model_path, map_location="cpu", weights_only=False)  # ✅ Fix PyTorch 2.6 issue
 
+# ✅ Initialize XTTS Model (CPU-only)
+tts = TTS(model_name="tts_models/multilingual/multi-dataset/xtts_v2").to("cpu")