rolpotamias commited on
Commit
5a6d9af
·
verified ·
1 Parent(s): 375318c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -33,7 +33,7 @@ model.eval()
33
 
34
  detector = YOLO('./pretrained_models/detector.pt').to(device)
35
 
36
- #@spaces.GPU#(duration=120)
37
  def run_wilow_model(image, conf, IoU_threshold=0.5):
38
  img_cv2 = image[...,::-1]
39
  img_vis = image.copy()
@@ -180,4 +180,4 @@ with gr.Blocks(title="WiLoR: End-to-end 3D hand localization and reconstruction
180
  ],
181
  inputs=input_image)
182
 
183
- demo.launch()
 
33
 
34
  detector = YOLO('./pretrained_models/detector.pt').to(device)
35
 
36
+ @spaces.GPU(enable_queue=True)
37
  def run_wilow_model(image, conf, IoU_threshold=0.5):
38
  img_cv2 = image[...,::-1]
39
  img_vis = image.copy()
 
180
  ],
181
  inputs=input_image)
182
 
183
+ demo.queue().launch(debug=True)