Update app.py

app.py

@@ -18,9 +18,9 @@ class_names = {
     5: "Copper",
 } if not model.names else model.names  # Use model.names if available
 
-def detect_pcb_faults(image):
-    """Runs YOLOv8 on the input image and returns detected defects."""
-    results = model(image, conf=0.18)  # 🔥 Lower confidence to detect more defects
+def detect_pcb_faults(image, conf_threshold):
+    """Runs YOLOv8 on the input image with an adjustable confidence threshold."""
+    results = model(image, conf=conf_threshold)  # 🔥 Adjustable confidence
 
     boxes = results[0].boxes.xyxy.cpu().numpy()  # Extract bounding boxes
     confs = results[0].boxes.conf.cpu().numpy()  # Extract confidence scores
@@ -38,11 +38,14 @@ def detect_pcb_faults(image):
 
     return image
 
-# ✅ Gradio UI for PCB Fault Detection
+# ✅ Gradio UI with Confidence Threshold Slider
 gr.Interface(
     fn=detect_pcb_faults,
-    inputs=gr.Image(type="numpy"),
+    inputs=[
+        gr.Image(type="numpy"),
+        gr.Slider(minimum=0.1, maximum=0.9, value=0.18, label="Confidence Threshold")
+    ],
     outputs=gr.Image(type="numpy"),
     title="PCB Fault Detection",
-    description="Upload a PCB image to detect defects using YOLOv8. The model will highlight and label detected faults."
+    description="Upload a PCB image to detect defects using YOLOv8. Adjust the confidence threshold to fine-tune detections."
 ).launch()