Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ def yolov8_inference(
|
|
30 |
|
31 |
image=image[:, :, ::-1].astype(np.uint8)
|
32 |
model = YOLO("https://huggingface.co/spaces/devisionx/Second_demo/blob/main/best.pt")
|
33 |
-
results = model(image,imgsz=320)[0]
|
34 |
image=image[:, :, ::-1].astype(np.uint8)
|
35 |
detections = sv.Detections.from_yolov8(results)
|
36 |
annotated_image = annotatormask.annotate(scene=image, detections=detections)
|
|
|
30 |
|
31 |
image=image[:, :, ::-1].astype(np.uint8)
|
32 |
model = YOLO("https://huggingface.co/spaces/devisionx/Second_demo/blob/main/best.pt")
|
33 |
+
results = model(image,imgsz=320,conf=conf_threshold,iou=iou_threshold)[0]
|
34 |
image=image[:, :, ::-1].astype(np.uint8)
|
35 |
detections = sv.Detections.from_yolov8(results)
|
36 |
annotated_image = annotatormask.annotate(scene=image, detections=detections)
|