Update app.py

app.py

@@ -1,4 +1,7 @@
+import os
 import gradio as gr
+from PIL import Image
+import pytesseract
 import yolov5
 
 # load model
@@ -11,10 +14,30 @@ model.agnostic = False  # NMS class-agnostic
 model.multi_label = False  # NMS multiple labels per box
 model.max_det = 1000  # maximum number of detections per image
 
-def greet(img):
+def license_plate_detect(img):
     results = model(img, size=640)
 
-    return "Hello " + str(results.pred) + "!!"
+    # parse results
+    if len(results.pred):
+        predictions = results.pred[0]
+        boxes = predictions[:, :4] # x1, y1, x2, y2
+        return boxes
+
+
+def read_license_number(img):
+    boxes = license_plate_detect(img)
+    if len(boxes[0]):
+        return [pytesseract.image_to_string(
+                    Image.crop(bbox.tolist()))
+               for bbox in boxes]
+
+
+def greet(img):
+    boxes = license_plate_detect(img)
+    return "Hello " + str([pytesseract.image_to_string(
+                    Image.crop(bbox.tolist()))
+               for bbox in boxes]) + "!!"
+
 
 iface = gr.Interface(fn=greet, inputs="image", outputs="text")
 iface.launch()