Update app.py

app.py

@@ -1,19 +1,49 @@
 import torch
 from ultralytics import YOLO
+import easyocr
+import cv2
+import numpy as np
 import gradio as gr
+import os
 
-# Trusted YOLO weights path
-WEIGHTS_PATH = "/app/yolov8n.pt"
+# Download pretrained ANPR model (trained to detect license plates)
+ANPR_WEIGHTS = "anpr_yolov8.pt"
+if not os.path.exists(ANPR_WEIGHTS):
+    os.system(f"wget -O {ANPR_WEIGHTS} https://github.com/madalinabuzatu/yolov8-license-plate-detection/releases/download/v1.0/best.pt")
 
-# Allow loading full model (Option 1 from error)
-torch.serialization.add_safe_globals([YOLO])
+# Load YOLO ANPR model
+model = YOLO(ANPR_WEIGHTS)
 
-model = YOLO(WEIGHTS_PATH, task="detect")
+# Load OCR reader
+reader = easyocr.Reader(['en'])
 
-def detect_objects(image):
+def detect_and_read_plate(image):
     results = model(image)
-    annotated = results[0].plot()
-    return annotated
+    for result in results:
+        boxes = result.boxes.xyxy.cpu().numpy()  # [x1, y1, x2, y2]
+        for box in boxes:
+            x1, y1, x2, y2 = map(int, box)
+            # Crop the detected license plate
+            plate_img = image[y1:y2, x1:x2]
+            if plate_img.size == 0:
+                continue
+            # OCR to read text
+            ocr_result = reader.readtext(plate_img)
+            if ocr_result:
+                text = " ".join([res[1] for res in ocr_result])
+                print(f"Detected Plate: {text}")
+                # Draw bounding box + text
+                cv2.rectangle(image, (x1, y1), (x2, y2), (0, 255, 0), 2)
+                cv2.putText(image, text, (x1, y1 - 10),
+                            cv2.FONT_HERSHEY_SIMPLEX, 0.8, (0, 255, 0), 2)
 
-demo = gr.Interface(fn=detect_objects, inputs="image", outputs="image")
+    return image
+
+demo = gr.Interface(
+    fn=detect_and_read_plate,
+    inputs="image",
+    outputs="image",
+    title="Automatic Number Plate Recognition (ANPR)",
+    description="Upload an image of a car to detect and read its license plate."
+)
 demo.launch(server_name="0.0.0.0", server_port=7860)