JunlinHan commited on
Commit
ddfecfc
1 Parent(s): 8a91955

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -178,7 +178,8 @@ def _default_render_cameras(batch_size: int = 1):
178
  render_camera_intrinsics = _default_intrinsics().unsqueeze(0).repeat(extrinsics.shape[0], 1, 1)
179
  render_cameras = _build_camera_standard(extrinsics, render_camera_intrinsics)
180
  return render_cameras.unsqueeze(0).repeat(batch_size, 1, 1)
181
-
 
182
  def generate_mesh(image, source_size=512, render_size=384, mesh_size=512, export_mesh=False, export_video=True, fps=30):
183
  image = preprocess_image(image, source_size).to(model_wrapper.device)
184
  source_camera = _default_source_camera(batch_size=1).to(model_wrapper.device)
 
178
  render_camera_intrinsics = _default_intrinsics().unsqueeze(0).repeat(extrinsics.shape[0], 1, 1)
179
  render_cameras = _build_camera_standard(extrinsics, render_camera_intrinsics)
180
  return render_cameras.unsqueeze(0).repeat(batch_size, 1, 1)
181
+
182
+ @spaces.GPU
183
  def generate_mesh(image, source_size=512, render_size=384, mesh_size=512, export_mesh=False, export_video=True, fps=30):
184
  image = preprocess_image(image, source_size).to(model_wrapper.device)
185
  source_camera = _default_source_camera(batch_size=1).to(model_wrapper.device)