Tharun156 commited on
Commit
efed32c
Β·
verified Β·
1 Parent(s): 18352ca

Update demo.py

Browse files
Files changed (1) hide show
  1. demo.py +7 -1
demo.py CHANGED
@@ -117,8 +117,14 @@ class BaseTrainer(object):
117
  logger.info(self.model)
118
  logger.info(f"init {cfg.model.g_name} success")
119
 
 
 
 
 
 
 
120
  self.smplx = smplx.SMPLX(
121
- model_path=self.args.data_path_1 + "smplx_models/",
122
  gender='NEUTRAL_2020',
123
  use_face_contour=False,
124
  num_betas=300,
 
117
  logger.info(self.model)
118
  logger.info(f"init {cfg.model.g_name} success")
119
 
120
+ smplx_path = Path(self.args.data_path_1) / "smplx_models"
121
+ if not smplx_path.exists():
122
+ raise FileNotFoundError(
123
+ "SMPL-X model directory missing at {}. Ensure assets are downloaded or"
124
+ " set HF_GESTURELSM_WEIGHTS_REPO with smplx_models.".format(smplx_path)
125
+ )
126
  self.smplx = smplx.SMPLX(
127
+ model_path=str(smplx_path),
128
  gender='NEUTRAL_2020',
129
  use_face_contour=False,
130
  num_betas=300,