Update lbw_detector.py

lbw_detector.py

@@ -4,10 +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)
+#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 = {