Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -31,6 +31,7 @@ def crop_images_from_boxes(image, boxes, scale_factor):
|
|
31 |
return cropped_images
|
32 |
|
33 |
# Warm-up function to perform a dummy inference
|
|
|
34 |
def warm_up_model():
|
35 |
dummy_image = np.zeros((640, 512, 3), dtype=np.uint8)
|
36 |
infer_image_and_get_boxes(dummy_image)
|
|
|
31 |
return cropped_images
|
32 |
|
33 |
# Warm-up function to perform a dummy inference
|
34 |
+
@spaces.GPU
|
35 |
def warm_up_model():
|
36 |
dummy_image = np.zeros((640, 512, 3), dtype=np.uint8)
|
37 |
infer_image_and_get_boxes(dummy_image)
|