allencbzhang commited on
Commit
6ab8d85
·
verified ·
1 Parent(s): cabf51c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -35,6 +35,7 @@ if __name__ == "__main__":
35
  metadata_dataset="coco_2017_val",
36
  )
37
 
 
38
  def inference(img, confidence):
39
  img = np.array(img)
40
  _, results = vis_demo.run_on_image(img, confidence)
 
35
  metadata_dataset="coco_2017_val",
36
  )
37
 
38
+ @spaces.GPU
39
  def inference(img, confidence):
40
  img = np.array(img)
41
  _, results = vis_demo.run_on_image(img, confidence)