Update main.py

main.py

@@ -6,6 +6,12 @@ import os
 from io import BytesIO
 from typing import Generator
 
+import torch
+from torch.serialization import add_safe_globals
+from TTS.tts.configs.xtts_config import XttsConfig
+
+# Allow this class to be unpickled (safe if you're sure the model is trusted)
+add_safe_globals([XttsConfig])