sayedM commited on
Commit
b3c2a3b
·
1 Parent(s): 2c46c5a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)