Update app.py

app.py

@@ -23,6 +23,9 @@ def detect_objects_on_image(image_path):
         "NP": "Number Plate"
     }
 
+    # Add some space around the text
+    text_padding = 5
+
     for box in result.boxes:
         x1, y1, x2, y2 = [round(x) for x in box.xyxy[0].tolist()]
         class_id = box.cls[0].item()
@@ -30,6 +33,12 @@ def detect_objects_on_image(image_path):
 
         class_name = class_names_mapping.get(result.names[class_id], result.names[class_id])
 
+        # Adjust the rectangle coordinates to add space around the text
+        x1 -= text_padding
+        y1 -= text_padding
+        x2 += text_padding
+        y2 += text_padding
+
         output.append([
             x1, y1, x2, y2, class_name, prob
         ])
@@ -48,6 +57,7 @@ def detect_objects_on_image(image_path):
     return cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
 
 
+
 inputs_image = [
     gr.components.Image(type="filepath", label="Input Image"),
 ]