Update app.py

app.py

@@ -332,13 +332,14 @@ def predict(image, model_path, class_names_path, confidence_threshold):
         return None, f"Error during inference: {e}", None
 
 
-def build_interface(model_path, class_names_path):
+def build_interface(model_path, class_names_path, example_images=None):
     """
     Build the Gradio interface components.
 
     Args:
         model_path: Path to the ONNX model
         class_names_path: Path to the class names file
+        example_images: List of example image paths
 
     Returns:
         gr.Blocks: The Gradio demo interface
@@ -374,6 +375,13 @@ def build_interface(model_path, class_names_path):
                         orientation="h",
                         label_title="Object Counts",
                     )
+        
+        # Add examples component if example images are provided
+        if example_images:
+            gr.Examples(
+                examples=example_images,
+                inputs=input_image,
+            )
 
         # Set up the click event inside the Blocks context
         submit_btn.click(
@@ -394,9 +402,25 @@ def launch_demo():
     """
     Launch the Gradio demo with hardcoded model and class names paths.
     """
-    demo = build_interface(MODEL_PATH, CLASS_NAMES_PATH)
-
-    # Launch the demo
+    # Create examples directory if it doesn't exist
+    examples_dir = os.path.join(BASE_DIR, "examples")
+    if not os.path.exists(examples_dir):
+        os.makedirs(examples_dir)
+        print(f"Created examples directory at {examples_dir}")
+    
+    # Get list of example images
+    example_images = []
+    if os.path.exists(examples_dir):
+        example_images = [
+            os.path.join(examples_dir, f) 
+            for f in os.listdir(examples_dir) 
+            if f.lower().endswith(('.png', '.jpg', '.jpeg'))
+        ]
+        print(f"Found {len(example_images)} example images")
+    
+    demo = build_interface(MODEL_PATH, CLASS_NAMES_PATH, example_images)
+    
+    # Launch the demo without the examples parameter
     demo.launch(share=False)  # Set share=True if you want to create a shareable link