track with git lfs and add examples

.gitattributes

@@ -33,3 +33,4 @@ 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
+*.png filter=lfs diff=lfs merge=lfs -text

app.py

@@ -17,7 +17,7 @@ model = AsymmetricCroCo3DStereo.from_pretrained(
 ).to(DEVICE)
 
 
-def create_blueprint(image_name_list: list[str], log_path: str) -> rrb.Blueprint:
+def create_blueprint(image_name_list: list[str], log_path: Path) -> rrb.Blueprint:
     # dont show 2d views if there are more than 4 images as to not clutter the view
     if len(image_name_list) > 4:
         blueprint = rrb.Blueprint(
@@ -87,13 +87,30 @@ with gr.Blocks(
 ) as demo:
     # scene state is save so that you can change conf_thr, cam_size... without rerunning the inference
     gr.HTML('<h2 style="text-align: center;">Mini-DUSt3R Demo</h2>')
+    gr.HTML(
+        '<p style="text-align: center;">Unofficial DUSt3R demo using the mini-dust3r pip package</p>'
+    )
+    gr.HTML(
+        '<p style="text-align: center;">More info <a href="https://github.com/pablovela5620/mini-dust3r">here</a></p>'
+    )
     with gr.Tab(label="Single Image"):
         with gr.Column():
-            singe_image = gr.Image(type="filepath")
+            single_image = gr.Image(type="filepath", height=300)
             run_btn_single = gr.Button("Run")
             rerun_viewer_single = Rerun(height=900)
             run_btn_single.click(
-                fn=predict, inputs=[singe_image], outputs=[rerun_viewer_single]
+                fn=predict, inputs=[single_image], outputs=[rerun_viewer_single]
+            )
+
+            example_single_dir = Path("examples/single_image")
+            example_single_files = sorted(example_single_dir.glob("*.png"))
+
+            examples_single = gr.Examples(
+                examples=example_single_files,
+                inputs=[single_image],
+                outputs=[rerun_viewer_single],
+                fn=predict,
+                cache_examples="lazy",
             )
     with gr.Tab(label="Multi Image"):
         with gr.Column():