Update app.py

app.py

@@ -33,7 +33,7 @@ def yolov8_inference(
 
     image=image[:, :, ::-1].astype(np.uint8)
     model = YOLO("https://huggingface.co/spaces/devisionx/Final_demo/blob/main/best_weights.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 = annotatormask.annotate(scene=image, detections=detections)
@@ -58,7 +58,7 @@ title = "Materials-Demo"
 '''
 import os
 examples = [
-    ["m1.jpg", 0.6, 0.45],
+    ["m1.jpg", 0.25, 0.45],
     ["m2.jpg", 0.25, 0.45],
     ["m3.jpg", 0.25, 0.45],
 ]