Upload app.py

app.py

@@ -430,9 +430,14 @@ def demo_lam(flametracking, lam, cfg):
                         template_vertices = flame_model.v_template.unsqueeze(0).expand(batch_size, -1, -1)
                         shapedirs = flame_model.shapedirs
 
-                    # shape blend
+                    # shape blend (shape_param이 1차원인 경우 2차원으로 변환)
                     n_shape_params = flame_model.n_shape_params if hasattr(flame_model, 'n_shape_params') else 10
-                    v_shaped = template_vertices + blend_shapes(shape_param, shapedirs[:, :, :n_shape_params])
+                    # shape_param을 올바른 차원으로 변환
+                    if shape_param.dim() == 1:
+                        shape_param_2d = shape_param.unsqueeze(0)  # (num_betas) -> (1, num_betas)
+                    else:
+                        shape_param_2d = shape_param
+                    v_shaped = template_vertices + blend_shapes(shape_param_2d, shapedirs[:, :, :n_shape_params])
 
                     # mesh 저장
                     mesh = trimesh.Trimesh(vertices=v_shaped.squeeze(0).cpu().numpy(), faces=faces)