Update app.py

app.py

@@ -193,47 +193,46 @@ with gr.Blocks(css=css, title="Real-Time YOLOv5 Video & Image Object Detection")
             )
             
         with gr.TabItem("Image Detection", elem_classes="tab-item"):
-    with gr.Row():
-        image_input = gr.Image(
-            type="pil", 
-            label="Upload Image", 
-            interactive=True
-        )
-    
-    with gr.Row(elem_classes="button-row"):  # Move button row here just below image_input
-        clear_button = gr.Button(
-            "Clear", 
-            variant="secondary", 
-            elem_id="clear-btn"
-        )
-        submit_button = gr.Button(
-            "Detect Objects", 
-            variant="primary", 
-            elem_id="submit-btn"
-        )
-    
-    # Define image_output below input and buttons
-    image_output = gr.Image(
-        label="Detected Objects", 
-        elem_id="image-output"
-    )
-    
-    clear_button.click(
-        fn=lambda: None, 
-        inputs=None, 
-        outputs=image_output
-    )
-    
-    submit_button.click(
-        fn=process_image, 
-        inputs=image_input, 
-        outputs=image_output
-    )
-
+            with gr.Row():
+                image_input = gr.Image(
+                    type="pil", 
+                    label="Upload Image", 
+                    interactive=True
+                )
+            
+            with gr.Row(elem_classes="button-row"):
+                clear_button = gr.Button(
+                    "Clear", 
+                    variant="secondary", 
+                    elem_id="clear-btn"
+                )
+                submit_button = gr.Button(
+                    "Detect Objects", 
+                    variant="primary", 
+                    elem_id="submit-btn"
+                )
+            
+            with gr.Row(elem_classes="output-container"):
+                image_output = gr.Image(
+                    label="Detected Objects", 
+                    elem_id="image-output"
+                )
+            
+            clear_button.click(
+                fn=lambda: None, 
+                inputs=None, 
+                outputs=image_output
+            )
+            
+            submit_button.click(
+                fn=process_image, 
+                inputs=image_input, 
+                outputs=image_output
+            )
     
     gr.Markdown("""
     ### Powered by YOLOv5.
     This application allows real-time object detection using the YOLOv5 model.
     """, elem_classes="footer")
 
-demo.launch()
+demo.launch()