Update services/detection_service.py

services/detection_service.py

@@ -6,4 +6,4 @@ object_detector = pipeline("object-detection", model="facebook/detr-resnet-50")
 def detect_objects(image_path):
     image = cv2.imread(image_path)
     results = object_detector(image)
-    return [r for r in results if r['score'] > 0.7 and r['label'] == 'pole']
+    return [r for r in results if r['score'] > 0.7]