Update app.py

app.py

@@ -1,19 +1,24 @@
-import gradio as gr
 import os
 import torch
-from TTS.tts.models.xtts import XttsAudioConfig  # ✅ Import the correct class
-from TTS.tts.configs.xtts_config import XttsConfig
+import gradio as gr
+from TTS.tts.configs.xtts_config import XttsConfig  # ✅ Import missing class
+from TTS.tts.models.xtts import XttsAudioConfig  # ✅ Import missing class
 from TTS.api import TTS
 
 # ✅ Accept Coqui License Automatically
 os.environ["COQUI_TOS_AGREED"] = "1"
 
 # ✅ Fix PyTorch deserialization issue
-torch.serialization.add_safe_globals([XttsAudioConfig])  # ✅ This fixes the issue
+torch.serialization.add_safe_globals([XttsConfig, XttsAudioConfig])  # ✅ Allow both classes
+
+# ✅ 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
 
-# Load the voice cloning model
-tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2").to("cpu")
 
 def generate_cloned_voice(text, reference_audio, language):
     output_path = "output.wav"