Spaces:
Runtime error
Runtime error
Commit
·
f7a089e
1
Parent(s):
6ac1fcf
Update app.py
Browse files
app.py
CHANGED
@@ -40,7 +40,7 @@ def yolov8_inference(
|
|
40 |
image=image[:, :, ::-1].astype(np.uint8)
|
41 |
model = YOLO("https://huggingface.co/spaces/devisionx/Fifth_demo/blob/main/best_weigh.pt")
|
42 |
|
43 |
-
results = model(image,
|
44 |
image=image[:, :, ::-1].astype(np.uint8)
|
45 |
detections = sv.Detections.from_yolov8(results)
|
46 |
annotated_image = annotatorbbox.annotate(scene=image, detections=detections)
|
|
|
40 |
image=image[:, :, ::-1].astype(np.uint8)
|
41 |
model = YOLO("https://huggingface.co/spaces/devisionx/Fifth_demo/blob/main/best_weigh.pt")
|
42 |
|
43 |
+
results = model(image,imgsz=360)[0]
|
44 |
image=image[:, :, ::-1].astype(np.uint8)
|
45 |
detections = sv.Detections.from_yolov8(results)
|
46 |
annotated_image = annotatorbbox.annotate(scene=image, detections=detections)
|