Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -33,7 +33,7 @@ model.eval()
|
|
33 |
|
34 |
detector = YOLO('./pretrained_models/detector.pt').to(device)
|
35 |
|
36 |
-
|
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)
|