Update app.py

app.py

@@ -65,11 +65,11 @@ def detect_objects(image):
         class_id = int(cls)
         color = colors[class_id].tolist()
         
-        # Thinner, stylish bounding boxes
-        cv2.rectangle(output_image, (x1, y1), (x2, y2), color, 2, lineType=cv2.LINE_AA)
+        # Thicker bounding boxes
+        cv2.rectangle(output_image, (x1, y1), (x2, y2), color, 3, lineType=cv2.LINE_AA)
         
         label = f"{model.names[class_id]} {conf:.2f}"
-        font_scale, font_thickness = 0.7, 1
+        font_scale, font_thickness = 0.9, 2  # Increased for better readability
         (w, h), _ = cv2.getTextSize(label, cv2.FONT_HERSHEY_SIMPLEX, font_scale, font_thickness)
         
         cv2.rectangle(output_image, (x1, y1 - h - 10), (x1 + w + 10, y1), color, -1)
@@ -89,7 +89,7 @@ def detect_objects(image):
     
     return output_image
 
-example_images = ["spring_street_after.jpg", "pexels-hikaique-109919.jpg"]
+examples = ["spring_street_after.jpg", "pexels-hikaique-109919.jpg"]
 os.makedirs("examples", exist_ok=True)
 
 with gr.Blocks(title="Optimized YOLOv5 Object Detection") as demo:
@@ -101,14 +101,12 @@ with gr.Blocks(title="Optimized YOLOv5 Object Detection") as demo:
     with gr.Row():
         with gr.Column(scale=1):
             input_image = gr.Image(label="Input Image", type="numpy")
+            submit_button = gr.Button("Submit", variant="primary")
+            clear_button = gr.Button("Clear")
         
         with gr.Column(scale=1):
             output_image = gr.Image(label="Detected Objects", type="numpy")
     
-    with gr.Row():
-        clear_button = gr.Button("Clear")
-        submit_button = gr.Button("Submit", variant="primary")
-    
     gr.Examples(
         examples=example_images,
         inputs=input_image,