add

app.py

@@ -79,13 +79,12 @@ ehm_model.head.load_state_dict(_state['head'], strict=False)
 ehm = EHM_v2("assets/FLAME", "assets/SMPLX", "assets/MANO")
 
 
-lights = PointLights(device=TORCH_DEVICE, location=[[0.0, -1.0, -10.0]])
+# lights = PointLights(device=TORCH_DEVICE, location=[[0.0, -1.0, -10.0]])
 
+# 2. 将 lights 的初始化移入函数，或确保它能动态转换
+def get_lights(device):
+    return PointLights(device=device, location=[[0.0, -1.0, -10.0]])
 
-if TORCH_DEVICE.type == "cuda":
-    body_renderer = body_renderer.cuda()
-    ehm_model = ehm_model.cuda()
-    ehm = ehm.cuda()
 
 
 
@@ -259,10 +258,15 @@ def handle_video_upload(video):
     return (json.dumps(frame_data)       )
 
 @spaces.GPU
-def mesh_inference( temp_dir, video_name):
-    """Run mesh inference on uploaded video"""
+def mesh_inference(temp_dir, video_name):
 
+    global body_renderer, ehm_model, ehm
+    body_renderer = body_renderer.to(TORCH_DEVICE)
+    ehm_model = ehm_model.to(TORCH_DEVICE)
+    ehm = ehm.to(TORCH_DEVICE)
+    lights = get_lights(TORCH_DEVICE)
 
+    # ... 后续推理逻辑 ...
     # Setup paths
     video_path = os.path.join(temp_dir, f"{video_name}.mp4")