Update coqui.py

coqui.py

@@ -9,9 +9,8 @@ import torch
 import torchaudio
 import numpy as np
 
-
-#update gradio to faster streaming
-#download for mecab
+# update gradio to faster streaming
+# download for mecab
 os.system('python -m unidic download')
 
 # By using XTTS you agree to CPML license https://coqui.ai/cpml
@@ -33,10 +32,12 @@ from TTS.api import TTS
 from TTS.tts.configs.xtts_config import XttsConfig
 from TTS.tts.models.xtts import Xtts
 from TTS.utils.generic_utils import get_user_data_dir
+from huggingface_hub import HfApi
 
 HF_TOKEN = os.environ.get("HF_TOKEN")
+if not HF_TOKEN:
+    raise ValueError("HF_TOKEN environment variable is not set")
 
-from huggingface_hub import HfApi
 # will use api to restart space on a unrecoverable error
 api = HfApi(token=HF_TOKEN)
 repo_id = "coqui/xtts"
@@ -54,15 +55,24 @@ config = XttsConfig()
 config.load_json(os.path.join(model_path, "config.json"))
 
 model = Xtts.init_from_config(config)
+checkpoint_path = os.path.join(model_path, "model.pth")
+vocab_path = os.path.join(model_path, "vocab.json")
+
+if not os.path.exists(checkpoint_path):
+    raise FileNotFoundError(f"Checkpoint file not found at {checkpoint_path}")
+if not os.path.exists(vocab_path):
+    raise FileNotFoundError(f"Vocab file not found at {vocab_path}")
+
 model.load_checkpoint(
     config,
-    checkpoint_path=os.path.join(model_path, "model.pth"),
-    vocab_path=os.path.join(model_path, "vocab.json"),
+    checkpoint_path=checkpoint_path,
+    vocab_path=vocab_path,
     eval=True,
     use_deepspeed=True,
 )
 model.cuda()
 
+
 # This is for debugging purposes only
 DEVICE_ASSERT_DETECTED = 0
 DEVICE_ASSERT_PROMPT = None