m

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+examples/*.png filter=lfs diff=lfs merge=lfs -text
+examples/*.jpg filter=lfs diff=lfs merge=lfs -text

app.py

@@ -95,11 +95,22 @@ def count_big_cats(img, score_threshold, iou_threshold):
     img_annotated = annotate(img.copy(), preds)
     return tiger_count, lion_count, total_count, img_annotated
 
+TEST_IMAGES = {
+    "Tiger": "examples/tiger1.png",
+    "Lion": "examples/tiger2.png",
+    "Both": "examples/tiger3.png",
+}
+
+def load_test_image(choice):
+    return Image.open(TEST_IMAGES[choice])
+
+
 with gr.Blocks(title="Big Cat Counter") as demo:
     gr.Markdown("# 🐯🦁 Big Cat Counter\nUpload an image and I’ll count how many **tigers** and **lions** I see.")
     with gr.Row():
         with gr.Column():
             inp = gr.Image(type="pil", label="Input image")
+            test_selector = gr.Dropdown(list(TEST_IMAGES.keys()), label="Pick a test image")
             score_th = gr.Slider(0.05, 0.95, value=0.20, step=0.05, label="Score threshold")
             iou_th = gr.Slider(0.1, 0.9, value=0.50, step=0.05, label="IOU (NMS) threshold")
             btn = gr.Button("Count Big Cats")
@@ -108,6 +119,7 @@ with gr.Blocks(title="Big Cat Counter") as demo:
             out_lion = gr.Number(label="Lion count", precision=0)
             out_total = gr.Number(label="Total big cats", precision=0)
             out_img = gr.Image(label="Annotated output")
+    test_selector.change(fn=load_test_image, inputs=test_selector, outputs=inp)
     btn.click(fn=count_big_cats, inputs=[inp, score_th, iou_th], outputs=[out_tiger, out_lion, out_total, out_img])
 
 if __name__ == "__main__":