Update app.py

app.py

@@ -36,10 +36,11 @@ def yolov8_inference(
 
     image=image[:, :, ::-1].astype(np.uint8)
     model = YOLO("https://huggingface.co/spaces/devisionx/Fourth_demo/blob/main/best.pt")
-    results = model(image,imgsz=640)[0]
+    results = model(image,imgsz=640,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)
 
 
 
@@ -61,7 +62,7 @@ title = "Wheel Segmentation Demo"
 '''
 import os
 examples = [
-    ["wh1.jpg", 0.6, 0.45],
+    ["wh1.jpg", 0.25, 0.45],
     ["wh2.jpg", 0.25, 0.45],
     ["wh3.jpg", 0.25, 0.45],
 ]