Spaces:
Running
on
Zero
Running
on
Zero
Add ZeroGPU decorator.
Browse files
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]
|