Update app.py

app.py

@@ -2,8 +2,6 @@ import gradio as gr
 import numpy as np
 from PIL import Image
 from ultralytics import YOLO
-import requests
-import os
 
 model = YOLO('best.pt')
 
@@ -30,10 +28,14 @@ def gallery_click_event(images, evt: gr.SelectData):
     selected_img = load_image_from_gallery(images, index)
     return selected_img
 
-with gr.Blocks() as demo:
+def clear_image():
+    return None
+
+with gr.Blocks(css=".container { background-color: white; }") as demo:
     with gr.Row():
         with gr.Column():
             selected_image = gr.Image(label="Selected Image from Gallery", type="pil")
+            clear_button = gr.Button("Clear")
         
         with gr.Column():
             image_gallery = gr.Gallery(label="Image Gallery", elem_id="gallery", type="pil")
@@ -41,18 +43,22 @@ with gr.Blocks() as demo:
         with gr.Column():
             result_image = gr.Image(label="Result Image", type="pil")
 
-    # Update selected image based on gallery click
     image_gallery.select(
         fn=gallery_click_event, 
         inputs=image_gallery, 
         outputs=selected_image
     )
     
-    # Predict and display the result image when the selected image changes
     selected_image.change(
         fn=predict, 
         inputs=selected_image, 
         outputs=result_image
     )
 
+    clear_button.click(
+        fn=clear_image,
+        inputs=None,
+        outputs=selected_image
+    )
+
 demo.launch()