JohanDL commited on
Commit
277dfb3
·
verified ·
1 Parent(s): aa766c3
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -11,9 +11,8 @@ import spaces
11
  from depth_anything.dpt import DepthAnything
12
  from depth_anything.util.transform import Resize, NormalizeImage, PrepareForNet
13
 
14
-
15
- @spaces.GPU
16
  @torch.no_grad()
 
17
  def predict_depth(model, image):
18
  return model(image)
19
 
 
11
  from depth_anything.dpt import DepthAnything
12
  from depth_anything.util.transform import Resize, NormalizeImage, PrepareForNet
13
 
 
 
14
  @torch.no_grad()
15
+ @spaces.GPU
16
  def predict_depth(model, image):
17
  return model(image)
18