Upload app.py

app.py

@@ -437,7 +437,13 @@ def demo_lam(flametracking, lam, cfg):
                         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])
+
+                    # 모든 텐서를 같은 디바이스로 통일 (CUDA)
+                    device = shape_param_2d.device
+                    template_vertices = template_vertices.to(device)
+                    shapedirs_subset = shapedirs[:, :, :n_shape_params].to(device)
+
+                    v_shaped = template_vertices + blend_shapes(shape_param_2d.to(device), shapedirs_subset)
 
                     # mesh 저장
                     mesh = trimesh.Trimesh(vertices=v_shaped.squeeze(0).cpu().numpy(), faces=faces)