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()
|
|
|
|
| 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()
|