update

app.py

@@ -120,13 +120,11 @@ with gr.Blocks() as demo:
             load_output = gr.Textbox(label="Model Status", interactive=False)
 
             gr.Examples(
-                examples=[
-                    os.path.join("examples", f)
-                    for f in os.listdir("examples")
-                ],
+                examples=[os.path.join("examples", f) for f in os.listdir("examples") if f.lower().endswith((".jpg", ".jpeg", ".png"))],
                 inputs=image_input,
                 label="Example Images",
-                examples_per_page=6,
+                examples_per_page=16,
+                elem_id="custom-example-gallery"
             )
         
         with gr.Column(scale=1):
@@ -137,6 +135,16 @@ with gr.Blocks() as demo:
             label_output = gr.Label(label="Prediction Probabilities")
             final_output = gr.Textbox(label="Final Prediction", interactive=False)
 
+        gr.HTML("""
+        <style>
+        #custom-example-gallery img {
+            width: 150px !important;
+            height: 150px !important;
+            object-fit: cover !important;
+        }
+        </style>
+        """)
+
     load_button.click(fn=load_model, inputs=model_selector, outputs=load_output)
     predict_button.click(fn=predict, inputs=image_input, outputs=[label_output, final_output])