More examples + cache

app.py

@@ -43,6 +43,9 @@ TORCH_DTYPE = torch.float32
 
 # Image
 IMAGE_EXAMPLES = [
+    {"path": "./examples/images/tennis.jpg", "use_url": False, "url": "", "label": "Local Image"},
+    {"path": "./examples/images/dogs.jpg", "use_url": False, "url": "", "label": "Local Image"},
+    {"path": "./examples/images/nascar.jpg", "use_url": False, "url": "", "label": "Local Image"},
     {"path": "./examples/images/crossroad.jpg", "use_url": False, "url": "", "label": "Local Image"},
     {
         "path": None,
@@ -59,6 +62,7 @@ ALLOWED_VIDEO_EXTENSIONS = {".mp4", ".avi", ".mov"}
 VIDEO_OUTPUT_DIR = Path("static/videos")
 VIDEO_OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
 VIDEO_EXAMPLES = [
+    {"path": "./examples/videos/dogs_running.mp4", "label": "Local Video"},
     {"path": "./examples/videos/traffic.mp4", "label": "Local Video"},
     {"path": "./examples/videos/fast_and_furious.mp4", "label": "Local Video"},
     {"path": "./examples/videos/break_dance.mp4", "label": "Local Video"},
@@ -129,8 +133,11 @@ def process_image(
     checkpoint: str = DEFAULT_CHECKPOINT,
     image: Optional[Image.Image] = None,
     url: Optional[str] = None,
+    use_url: bool = False,
     confidence_threshold: float = DEFAULT_CONFIDENCE_THRESHOLD,
 ):
+    if not use_url:
+        url = None
 
     if (image is None) ^ bool(url):
         raise ValueError(f"Either image or url must be provided, but not both.")
@@ -344,25 +351,26 @@ with gr.Blocks(theme=gr.themes.Ocean()) as demo:
             gr.Examples(
                 examples=[
                     [
+                        DEFAULT_CHECKPOINT,
                         example["path"],
-                        example["use_url"],
                         example["url"],
-                        DEFAULT_CHECKPOINT,
+                        example["use_url"],
                         DEFAULT_CONFIDENCE_THRESHOLD,
                     ]
                     for example in IMAGE_EXAMPLES
                 ],
                 inputs=[
+                    image_model_checkpoint,
                     image_input,
-                    use_url,
                     url_input,
-                    image_model_checkpoint,
+                    use_url,
                     image_confidence_threshold,
                 ],
                 outputs=[image_output],
                 fn=process_image,
-                cache_examples=False,
                 label="Select an image example to populate inputs",
+                cache_examples=True,
+                cache_mode="lazy",
             )
 
         with gr.Tab("Video"):
@@ -389,7 +397,8 @@ with gr.Blocks(theme=gr.themes.Ocean()) as demo:
                 inputs=[video_input, video_checkpoint, video_confidence_threshold],
                 outputs=[video_output],
                 fn=process_video,
-                cache_examples=False,
+                cache_examples=True,
+                cache_mode="lazy",
                 label="Select a video example to populate inputs",
             )
 
@@ -443,6 +452,7 @@ with gr.Blocks(theme=gr.themes.Ocean()) as demo:
             image_model_checkpoint,
             image_input,
             url_input,
+            use_url,
             image_confidence_threshold,
         ],
         outputs=[image_output],

examples/videos/dogs_running.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:043c1a28bddcd9beeddbab946c38d1a91de2a18da3ae114d307cd01b30b64ca2
+size 22119043