Update services/detection_service.py

services/detection_service.py

@@ -1,11 +1,9 @@
 from transformers import pipeline
 import cv2
 
-# Load DETR model
 object_detector = pipeline("object-detection", model="facebook/detr-resnet-50")
 
 def detect_objects(image_path):
     image = cv2.imread(image_path)
     results = object_detector(image)
-    high_conf = [r for r in results if r['score'] > 0.8]
-    return high_conf
+    return [r for r in results if r['score'] > 0.7 and r['label'] == 'pole']