Update app.py

app.py

@@ -14,6 +14,7 @@ from functools import partial
 from kiui.op import recenter
 import kiui
 from gradio_litmodel3d import LitModel3D
+import spaces
 
 # we load the pre-trained model from HF
 class LRMGeneratorWrapper:
@@ -152,7 +153,7 @@ def _default_render_cameras(batch_size: int = 1):
     render_camera_intrinsics = _default_intrinsics().unsqueeze(0).repeat(extrinsics.shape[0], 1, 1)
     render_cameras = _build_camera_standard(extrinsics, render_camera_intrinsics)
     return render_cameras.unsqueeze(0).repeat(batch_size, 1, 1)
-
+@spaces.GPU()
 def generate_mesh(image, source_size=512, render_size=384, mesh_size=512, export_mesh=False, export_video=True, fps=30):
     image = preprocess_image(image, source_size).to(model_wrapper.device)
     source_camera = _default_source_camera(batch_size=1).to(model_wrapper.device)