Update app.py

app.py

@@ -1,15 +1,16 @@
 import os
 import torch
 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.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.api import TTS
 
 # ✅ Accept Coqui License Automatically
 os.environ["COQUI_TOS_AGREED"] = "1"
 
-# ✅ Fix PyTorch deserialization issue
-torch.serialization.add_safe_globals([XttsConfig, XttsAudioConfig])  # ✅ Allow both classes
+# ✅ Allow missing classes for safe deserialization
+torch.serialization.add_safe_globals([XttsConfig, XttsAudioConfig, BaseDatasetConfig])
 
 # ✅ Force full checkpoint loading
 def safe_load_checkpoint(model_path):
@@ -20,6 +21,7 @@ model_name = "tts_models/multilingual/multi-dataset/xtts_v2"
 tts = TTS(model_name=model_name).to("cpu")  # ✅ Ensure CPU usage
 
 
+
 def generate_cloned_voice(text, reference_audio, language):
     output_path = "output.wav"