Update dataloaders/beat_sep_single.py

dataloaders/beat_sep_single.py

@@ -19,6 +19,8 @@ import pickle
 import librosa
 import smplx
 
+from pathlib import Path
+
 from .build_vocab import Vocab
 from models.utils.wav2vec import Wav2Vec2Model
 from .data_tools import joints_list
@@ -76,8 +78,13 @@ class CustomDataset(Dataset):
                 else:
                     self.joint_mask[self.ori_joint_list[joint_name][1] - self.ori_joint_list[joint_name][0]:self.ori_joint_list[joint_name][1]] = 1
         # select trainable joints
+        smplx_path = Path(self.args.data_path_1) / "smplx_models"
+        if not smplx_path.exists():
+            raise FileNotFoundError(
+                f"SMPL-X assets missing at {smplx_path}. Ensure they are available before running the demo."
+            )
         self.smplx = smplx.SMPLX(
-            model_path=self.args.data_path_1 + "smplx_models/",
+            model_path=str(smplx_path),
             gender='NEUTRAL_2020',
             use_face_contour=False,
             num_betas=300,