Update app.py

torch.serialization.add_safe_globals([XttsConfig, XttsAudioConfig])

app.py

@@ -9,6 +9,7 @@ from TTS.api import TTS
 from TTS.utils.manage import ModelManager
 # attempt to import torch.serilization to fix current runtime error
 from TTS.tts.configs.xtts_config import XttsConfig
+from TTS.tts.models.xtts import XttsAudioConfig
 import torch.serialization
 # torch.serialization.add_safe_globals([XttsConfig])
 model_names = TTS().list_models()
@@ -19,10 +20,10 @@ model_name = "tts_models/multilingual/multi-dataset/xtts_v2" # move in v2, since
 #m = ModelManager().download_model(model_name)
 #print(m)
 m = model_name
-with torch.serialization.safe_globals([XttsConfig]):
-    tts = TTS(model_name, gpu=False)
-    tts.to("cpu") # no GPU or Amd
-    #tts.to("cuda") # cuda only
+torch.serialization.add_safe_globals([XttsConfig, XttsAudioConfig])
+tts = TTS(model_name, gpu=False)
+tts.to("cpu") # no GPU or Amd
+#tts.to("cuda") # cuda only
 
 
 def predict(prompt, language, audio_file_pth, mic_file_path, use_mic, agree, request: gr.Request):