Update app.py

app.py

@@ -37,10 +37,10 @@ def yolov8_inference(
    
     image=image[:, :, ::-1].astype(np.uint8)
     model = YOLO("https://huggingface.co/spaces/devisionx/Fifth_demo/blob/main/best_weigh.pt")
-    results = model(image,imgsz=360)[0]
+    results = model(image,imgsz=360,conf=conf_threshold,iou=iou_threshold)[0]
     image=image[:, :, ::-1].astype(np.uint8)
     detections = sv.Detections.from_yolov8(results)
-    annotated_image = annotatorbbox.annotate(scene=image, detections=detections)
+    annotated_image = annotatormask.annotate(scene=image, detections=detections)
     annotated_image = annotatorbbox.annotate(scene=annotated_image , detections=detections)