mennamostafa55555 commited on
Commit
f7a089e
·
1 Parent(s): 6ac1fcf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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,conf=conf_threshold,iou=iou_threshold ,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)
 
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)