Update lbw_detector.py

lbw_detector.py

@@ -4,7 +4,10 @@ import numpy as np
 import os
 
 # Load YOLO model (custom-trained or pretrained with compatible classes)
-model = YOLO("yolov8n.pt")  # Replace with "lbw_yolov8.pt" if custom-trained
+#model = YOLO("yolov8n.pt")  # Replace with "lbw_yolov8.pt" if custom-trained
+
+model_path = os.path.join("models", "yolov8n.pt")
+model = YOLO(model_path)
 
 # Target class IDs — update based on your custom model class mapping
 CLASS_NAMES = {