Spaces:
Sleeping
Sleeping
Commit ·
5520222
1
Parent(s): f017173
Update app.py
Browse files
app.py
CHANGED
|
@@ -35,7 +35,7 @@ def yolov8_inference(
|
|
| 35 |
|
| 36 |
model = YOLO("https://huggingface.co/spaces/devisionx/Amazon_demo/blob/main/amazon.pt")
|
| 37 |
|
| 38 |
-
results = model(image,conf=conf_threshold,iou=iou_threshold ,imgsz=1280
|
| 39 |
detections = sv.Detections.from_yolov8(results)
|
| 40 |
annotated_image = annotatorbbox.annotate(scene=image, detections=detections)
|
| 41 |
annotated_image = annotatormask.annotate(scene=annotated_image, detections=detections)
|
|
|
|
| 35 |
|
| 36 |
model = YOLO("https://huggingface.co/spaces/devisionx/Amazon_demo/blob/main/amazon.pt")
|
| 37 |
|
| 38 |
+
results = model(image,conf=conf_threshold,iou=iou_threshold ,imgsz=1280)[0]
|
| 39 |
detections = sv.Detections.from_yolov8(results)
|
| 40 |
annotated_image = annotatorbbox.annotate(scene=image, detections=detections)
|
| 41 |
annotated_image = annotatormask.annotate(scene=annotated_image, detections=detections)
|