Spaces:
Running
Running
Update app.py
Browse files
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)
|