Update app.py

app.py

@@ -10,6 +10,9 @@ model = YOLO("best.pt")
 CLASS_NAMES  = {0: "Full",   1: "Broken"}
 CLASS_COLORS = {0: (34, 197, 94), 1: (239, 68, 68)}   # green, red
 
+# ── Sample images ─────────────────────────────────────────────────────────────
+SAMPLE_PATHS = ["image1.jpg", "image2.jpg"]
+
 
 # ── Inference ─────────────────────────────────────────────────────────────────
 def predict(image: Image.Image):
@@ -285,7 +288,7 @@ with gr.Blocks(css=css, title="GrainVision — Rice Grain Classifier") as demo:
     </div>
     """)
 
-    with gr.Row(equal_height=False):
+    with gr.Row(equal_height=True):
 
         # ── Left: Input ───────────────────────────────────────────────────────
         with gr.Column(scale=1):
@@ -294,10 +297,17 @@ with gr.Blocks(css=css, title="GrainVision — Rice Grain Classifier") as demo:
                 type="pil",
                 label="",
                 elem_classes=["image-wrap"],
-                height=420,
+                height=400,
             )
-            gr.HTML('<div style="height:12px"></div>')
+            gr.HTML('<div style="height:10px"></div>')
             run_btn = gr.Button("Analyse Grains", elem_id="run-btn")
+            gr.HTML('<div class="section-label" style="margin-top:16px;">Try an Example</div>')
+            gr.Examples(
+                examples=[[p] for p in SAMPLE_PATHS],
+                inputs=input_img,
+                label="",
+                examples_per_page=2,
+            )
 
         # ── Right: Output ─────────────────────────────────────────────────────
         with gr.Column(scale=1):
@@ -306,7 +316,7 @@ with gr.Blocks(css=css, title="GrainVision — Rice Grain Classifier") as demo:
                 type="pil",
                 label="",
                 elem_classes=["image-wrap"],
-                height=420,
+                height=400,
             )
 
     gr.HTML('<div style="height:20px"></div>')