Spaces:
Runtime error
Runtime error
Update demo.py
Browse files
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=
|
| 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,
|