Update app.py

app.py

@@ -12,8 +12,8 @@ BOX_COLORS = {
 BOX_PADDING = 2
 
 # Load models
-DETECTION_MODEL = YOLO("models/detector-model.pt") 
-CLASSIFICATION_MODEL = YOLO("models/classifier-model.pt") # 0: block, 1: checked, 2: unchecked
+DETECTION_MODEL = YOLO("detector-model.pt") 
+CLASSIFICATION_MODEL = YOLO("classifier-model.pt") # 0: block, 1: checked, 2: unchecked
 
 def detect(image_path):
     """
@@ -29,7 +29,7 @@ def detect(image_path):
         return image
     
     # Predict on image
-    results = DETECTION_MODEL.predict(source=image, conf=0.2, iou=0.8) # Predict on image
+    results = DETECTION_MODEL.predict(source=image, conf=0.5, iou=0.8) # Predict on image
     boxes = results[0].boxes # Get bounding boxes
 
     if len(boxes) == 0: