Update app.py

app.py

@@ -1,10 +1,24 @@
+import os
 import gradio as gr
 import numpy as np
 from PIL import Image
 from ultralytics import YOLO
 
+# Load the model
 model = YOLO('best.pt')
 
+# Path to the photos folder
+photos_folder = "Photos"
+
+def load_images_from_folder(folder):
+    images = []
+    for filename in os.listdir(folder):
+        if filename.lower().endswith(('.png', '.jpg', '.jpeg')):
+            img_path = os.path.join(folder, filename)
+            img = Image.open(img_path)
+            images.append((img, filename))
+    return images
+
 def predict(image):
     try:
         image = np.array(image)
@@ -31,6 +45,9 @@ def gallery_click_event(images, evt: gr.SelectData):
 def clear_image():
     return None
 
+# Load images at the start
+images = load_images_from_folder(photos_folder)
+
 with gr.Blocks(css=".container { background-color: white; }") as demo:
     with gr.Row():
         with gr.Column():
@@ -38,7 +55,7 @@ with gr.Blocks(css=".container { background-color: white; }") as demo:
             clear_button = gr.Button("Clear")
         
         with gr.Column():
-            image_gallery = gr.Gallery(label="Image Gallery", elem_id="gallery", type="pil")
+            image_gallery = gr.Gallery(label="Image Gallery", elem_id="gallery", type="pil", value=[img for img, _ in images])
         
         with gr.Column():
             result_image = gr.Image(label="Result Image", type="pil")