Update app.py

app.py

@@ -1,4 +1,5 @@
 import io
+import os
 import torch
 import numpy as np
 import gradio as gr
@@ -85,18 +86,14 @@ def fig2img(fig):
 description = """Upload an image and get the predicted classes"""
 title = """Object Detection"""
 
-classes_to_show = gr.components.Textbox(
-    placeholder="e.g. car, dog",
-    label="Classes to filter (leave empty to detect all classes)",
-)
-prob_threshold_slider = gr.components.Slider(minimum=0, maximum=1.0, step=0.01, value=0.7, label="Probability Threshold")
+# Create examples list from "examples/" directory
+example_list = [["examples/" + example] for example in os.listdir("examples")]
+
 
 image_in = gr.components.Image(label="Upload an image")
 image_out = gr.components.Image()
-
-
-
-
+classes_to_show = gr.components.Textbox(placeholder="e.g. car, dog", label="Classes to filter (leave empty to detect all classes)")
+prob_threshold_slider = gr.components.Slider(minimum=0, maximum=1.0, step=0.01, value=0.7, label="Probability Threshold")
 inputs = [image_in, prob_threshold_slider, classes_to_show]
 
 gr.Interface(fn=model_inference,
@@ -104,5 +101,5 @@ gr.Interface(fn=model_inference,
              outputs=image_out,
              title=title,
              description=description,
-             # examples=["CTH.png", "carplane.webp"]
+             examples=example_list
 ).launch()