Update services/road_safety/barrier_check.py

services/road_safety/barrier_check.py

@@ -5,7 +5,7 @@ import os
 
 # Load YOLOv8m model
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
-MODEL_PATH = os.path.join(BASE_DIR, "../../models/yolov8m.pt")
+MODEL_PATH = os.path.abspath(os.path.join(BASE_DIR, "../../models/yolov8m.pt"))
 model = YOLO(MODEL_PATH)
 
 def process_barriers(frame):
@@ -15,7 +15,7 @@ def process_barriers(frame):
         frame: Input frame (numpy array)
     Returns:
         list: List of detected barriers
-        numpy array: Annotated frame
+        numpy array: Annotated frame with numbered labels
     """
     results = model(frame)