Update app.py

app.py

@@ -34,6 +34,24 @@ except Exception as e:
 st.title("YOLOv11 Object Detection")
 st.write("Upload an image and let the model detect objects.")
 
+# Sliders for Confidence and Overlap thresholds
+confidence_threshold = st.slider(
+    "Confidence Threshold",
+    min_value=0.0,
+    max_value=1.0,
+    value=0.25,
+    step=0.01,
+    help="Set the minimum confidence score for detections.",
+)
+overlap_threshold = st.slider(
+    "Overlap Threshold",
+    min_value=0.0,
+    max_value=1.0,
+    value=0.45,
+    step=0.01,
+    help="Set the maximum allowable overlap for non-max suppression.",
+)
+
 uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "png", "jpeg"])
 if uploaded_file:
     # Read and display the image
@@ -43,7 +61,11 @@ if uploaded_file:
     # Perform prediction
     with st.spinner("Processing..."):
         try:
-            results = model.predict(np.array(image))
+            results = model.predict(
+                np.array(image),
+                conf=confidence_threshold,  # Set confidence threshold
+                iou=overlap_threshold,  # Set overlap (IoU) threshold
+            )
             st.write("Detection Results:")
             st.image(results[0].plot(), caption="Detections", use_column_width=True)
         except Exception as e: