Update app.py

fixing runtime errors

app.py

@@ -7,9 +7,10 @@ os.environ["COQUI_TOS_AGREED"] = "1"
 import gradio as gr
 from TTS.api import TTS
 from TTS.utils.manage import ModelManager
-# attempt to import torch.serilization to fix current runtime error
+# fixing current runtime errors
 from TTS.tts.configs.xtts_config import XttsConfig
 from TTS.tts.models.xtts import XttsAudioConfig
+from TTS.config.shared_configs import BaseDatasetConfig
 import torch.serialization
 # torch.serialization.add_safe_globals([XttsConfig])
 model_names = TTS().list_models()
@@ -20,7 +21,7 @@ model_name = "tts_models/multilingual/multi-dataset/xtts_v2" # move in v2, since
 #m = ModelManager().download_model(model_name)
 #print(m)
 m = model_name
-torch.serialization.add_safe_globals([XttsConfig, XttsAudioConfig])
+torch.serialization.add_safe_globals([XttsConfig, XttsAudioConfig, BaseDatasetConfig])
 tts = TTS(model_name, gpu=False)
 tts.to("cpu") # no GPU or Amd
 #tts.to("cuda") # cuda only