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/Amazon_demo/blob/main/amazon.pt")
|
| 33 |
-
|
| 34 |
results = model(image,conf=conf_threshold,iou=iou_threshold ,imgsz=1280)[0]
|
| 35 |
image=image[:, :, ::-1].astype(np.uint8)
|
| 36 |
detections = sv.Detections.from_yolov8(results)
|
|
|
|
| 30 |
|
| 31 |
image=image[:, :, ::-1].astype(np.uint8)
|
| 32 |
model = YOLO("https://huggingface.co/spaces/devisionx/Amazon_demo/blob/main/amazon.pt")
|
| 33 |
+
print("conf_threshold : ",conf_threshold ," iou_threshold : ",iou_threshold)
|
| 34 |
results = model(image,conf=conf_threshold,iou=iou_threshold ,imgsz=1280)[0]
|
| 35 |
image=image[:, :, ::-1].astype(np.uint8)
|
| 36 |
detections = sv.Detections.from_yolov8(results)
|