add cached examples

gradio_demo.py

@@ -88,7 +88,7 @@ def log_images_segmentation(
 
 
 # Gradio stuff
-def predict():
+def predict(prompt: str, image_path: str):
     file_name = f"{datetime.utcnow().strftime('%s')}.html"
     file_path = static_dir / file_name
     rec = rr.memory_recording()
@@ -102,26 +102,22 @@ def predict():
 
     predictor = SamPredictor(sam)
 
-    # prompt = "tires"
+    # log_video_segmentation(
+    #     Path("dog_and_woman.mp4"),
+    #     "dog, woman",
+    #     model,
+    #     predictor,
+    #     device=device,
+    # )
 
-    log_video_segmentation(
-        Path("dog_and_woman.mp4"),
-        "dog, woman",
+    log_images_segmentation(
+        [image_path],
+        prompt,
         model,
         predictor,
         device=device,
     )
 
-    # log_images_segmentation(
-    #     [
-    #         "https://raw.githubusercontent.com/facebookresearch/segment-anything/main/notebooks/images/truck.jpg"
-    #     ],
-    #     prompt,
-    #     model,
-    #     predictor,
-    #     device=device,
-    # )
-
     with open(file_path, "w") as f:
         f.write(rec.as_html())
     iframe = f"""<iframe src="/static/{file_name}" width="950" height="712"></iframe>"""
@@ -130,8 +126,25 @@ def predict():
 
 with gr.Blocks() as block:
     with gr.Row():
-        html = gr.HTML(label="HTML preview", show_label=True, value=predict())
+        with gr.Column():
+            text_input = gr.Textbox(value="tires, wheels", label="Prompt")
+            new_btn = gr.Button("Segment")
+        with gr.Column():
+            image = gr.Image(label="Image", type="filepath")
+    with gr.Row():
+        html = gr.HTML(label="HTML preview", show_label=True)
+    with gr.Row():
+        gr.Examples(
+            [
+                ["fan, mirror, sofa", "living_room.jpeg"],
+            ],
+            fn=predict,
+            inputs=[text_input, image],
+            outputs=[html],
+            cache_examples=True,
+        )
 
+    new_btn.click(fn=predict, inputs=[text_input, image], outputs=[html])
 # mount Gradio app to FastAPI app
 app = gr.mount_gradio_app(app, block, path="/")