Hancy commited on
Commit
062d491
Β·
verified Β·
1 Parent(s): 2014fad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ def colorize_depth(depth, log_scale):
29
  rgb[mask] = 0.
30
  return rgb
31
 
32
-
33
  @torch.no_grad()
34
  def generate_lidar(model, cond):
35
  img, pcd = sample_cond.sample(model, cond)
 
29
  rgb[mask] = 0.
30
  return rgb
31
 
32
+ @spaces.GPU
33
  @torch.no_grad()
34
  def generate_lidar(model, cond):
35
  img, pcd = sample_cond.sample(model, cond)