ZhengPeng7 commited on
Commit
2676e90
·
verified ·
1 Parent(s): 614027e

Add ZeroGPU decorator.

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -34,6 +34,7 @@ if os.path.exists(state_dict):
34
  model.eval()
35
 
36
 
 
37
  def pred_maps(image_1, image_2, image_3, image_4):
38
  images = [image_1, image_2, image_3, image_4]
39
  image_shapes = [image.shape[:2] for image in images]
 
34
  model.eval()
35
 
36
 
37
+ @spaces.GPU
38
  def pred_maps(image_1, image_2, image_3, image_4):
39
  images = [image_1, image_2, image_3, image_4]
40
  image_shapes = [image.shape[:2] for image in images]