zliang commited on
Commit
6d9fcdf
·
verified ·
1 Parent(s): 0bd2433

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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)