Update app.py

app.py

@@ -15,7 +15,7 @@ reader = easyocr.Reader(['en'])
 # Directory to save images of non-helmet riders
 os.makedirs("non_helmet_riders", exist_ok=True)
 
-# Function to enhance the image for better number plate recognition
+# Function to preprocess image for better OCR performance
 def preprocess_image_for_ocr(image):
     gray = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2GRAY)
     _, thresh = cv2.threshold(gray, 150, 255, cv2.THRESH_BINARY)
@@ -27,13 +27,13 @@ def detect_non_helmet_and_plate(image):
     results = model(image)
 
     helmet_status = "Pass"
-    license_plate_text = "I can't detect image"
+    license_plate_text = "No license plate detected"
     license_plate_image = None
-
     non_helmet_detected = False
+
     for *xyxy, conf, cls in results.xyxy[0]:
         class_id = int(cls)
-        if class_id == 0:  # Class 0 is 'person' in YOLOv5s
+        if class_id == 0:  # Class 0 is 'person'
             non_helmet_detected = True
             helmet_status = "Fail"
             cv2.rectangle(img_np, (int(xyxy[0]), int(xyxy[1])),