Update coqui.py

coqui.py

@@ -51,6 +51,10 @@ ModelManager().download_model(model_name)
 model_path = os.path.join(get_user_data_dir("tts"), model_name.replace("/", "--"))
 print("XTTS downloaded")
 
+# Ensure the model directory and files have the correct permissions
+if not os.access(model_path, os.W_OK):
+    raise PermissionError(f"Write permission denied for model directory: {model_path}")
+
 config = XttsConfig()
 config.load_json(os.path.join(model_path, "config.json"))
 
@@ -58,11 +62,10 @@ model = Xtts.init_from_config(config)
 checkpoint_path = os.path.join(model_path, "model.pth")
 vocab_path = os.path.join(model_path, "vocab.json")
 
-print("config : ",config)
 if not os.path.exists(checkpoint_path):
-    print(f"Checkpoint file not found at {checkpoint_path}")
+    raise FileNotFoundError(f"Checkpoint file not found at {checkpoint_path}")
 if not os.path.exists(vocab_path):
-    print(f"Vocab file not found at {vocab_path}")
+    raise FileNotFoundError(f"Vocab file not found at {vocab_path}")
 
 model.load_checkpoint(
     config,
@@ -73,7 +76,6 @@ model.load_checkpoint(
 )
 model.cuda()
 
-
 # This is for debugging purposes only
 DEVICE_ASSERT_DETECTED = 0
 DEVICE_ASSERT_PROMPT = None