add 3 examples

app.py

@@ -57,7 +57,6 @@ VARIANT_PREFIX = {
 
 # ——— Helper: download checkpoint if remote ———
 def get_checkpoint(path_or_key: str) -> str:
-    # If it's a key in REMOTE_CHECKPOINTS, download
     if path_or_key in REMOTE_CHECKPOINTS:
         url = REMOTE_CHECKPOINTS[path_or_key]
         local_path = f"/tmp/{path_or_key}.pth"
@@ -67,7 +66,6 @@ def get_checkpoint(path_or_key: str) -> str:
                 for chunk in r.iter_content(1024):
                     f.write(chunk)
         return local_path
-    # Otherwise assume it's a local file path
     return path_or_key
 
 # ——— Detect variant alias from checkpoint ———
@@ -88,34 +86,26 @@ def load_inferencer(checkpoint_path=None, device=None):
         kwargs['pose2d'] = variant
         kwargs['pose2d_weights'] = checkpoint_path
     else:
-        # default to rtmo-s
         kwargs['pose2d'] = 'rtmo'
     return MMPoseInferencer(**kwargs)
 
-# ——— Gradio prediction function ———
+# —─── Prediction function ────
 @spaces.GPU()
 def predict(image: Image.Image,
             remote_ckpt: str,
             upload_ckpt,
             bbox_thr: float,
             nms_thr: float):
-    # save input image
     inp_path = "/tmp/upload.jpg"
     image.save(inp_path)
-
-    # choose checkpoint: upload overrides remote
     if upload_ckpt:
         ckpt_path = upload_ckpt.name
         active = os.path.basename(ckpt_path)
     else:
         ckpt_path = get_checkpoint(remote_ckpt)
         active = remote_ckpt
-
-    # prepare output dir
     vis_dir = "/tmp/vis"
     os.makedirs(vis_dir, exist_ok=True)
-
-    # run inference
     inferencer = load_inferencer(checkpoint_path=ckpt_path, device=None)
     for result in inferencer(
         inputs=inp_path,
@@ -126,14 +116,11 @@ def predict(image: Image.Image,
         vis_out_dir=vis_dir,
     ):
         pass
-
-    # load result image
     out_files = sorted(os.listdir(vis_dir))
     vis_img = Image.open(os.path.join(vis_dir, out_files[0])) if out_files else None
     return vis_img, active
 
-# Build Gradio UI with Blocks for improved layout
-
+# —─── Gradio UI ────
 def main():
     with gr.Blocks() as demo:
         gr.Markdown("## RTMO Pose Demo")
@@ -153,6 +140,25 @@ def main():
                 output_img = gr.Image(type="pil", label="Annotated Image",
                                       elem_id="output_image", interactive=False)
                 active_tb = gr.Textbox(label="Active Checkpoint", interactive=False)
+        
+        # Examples for quick testing
+        gr.Examples(
+            examples=[
+                ["https://images.pexels.com/photos/1858175/pexels-photo-1858175.jpeg?auto=compress&cs=tinysrgb&dpr=1&w=614",
+                 "rtmo-s_coco_retrainable", None, 0.1, 0.65],
+                ["https://images.pexels.com/photos/3779706/pexels-photo-3779706.jpeg?auto=compress&cs=tinysrgb&dpr=1&w=614",
+                 "rtmo-t_8xb32-600e_body7", None, 0.1, 0.65],
+                ["https://images.pexels.com/photos/220453/pexels-photo-220453.jpeg?auto=compress&cs=tinysrgb&dpr=1&w=614",
+                 "rtmo-s_8xb32-600e_coco", None, 0.1, 0.65],
+            ],
+            inputs=[img_input, remote_dd, upload_ckpt, bbox_thr, nms_thr],
+            outputs=[output_img, active_tb],
+            fn=predict,
+            cache_examples=False,
+            label="Examples",
+            examples_per_page=3
+        )
+
         run_btn.click(predict,
                       inputs=[img_input, remote_dd, upload_ckpt, bbox_thr, nms_thr],
                       outputs=[output_img, active_tb])