Update services/road_safety/pothole_crack_detection.py

services/road_safety/pothole_crack_detection.py

@@ -21,6 +21,8 @@ MODEL_PATH = os.path.abspath(os.path.join(BASE_DIR, "../../models/yolov8m-seg.pt
 try:
     model = YOLO(MODEL_PATH)
     logging.info("Successfully loaded YOLOv8m-seg model for pothole and crack detection.")
+    # Log the model's class names for debugging
+    logging.info(f"Model class names: {model.names}")
 except Exception as e:
     logging.error(f"Failed to load YOLOv8m-seg model: {str(e)}")
     model = None