services/detection_service.py

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