feat: add example

app.py

@@ -5,6 +5,12 @@ import numpy as np
 from depth_pro.depth_pro import DepthProConfig
 from PIL import Image
 
+MARKDOWN = """
+        <div align="center">
+            <h2><a href="https://arxiv.org/abs/2410.02073">Depth Pro: Sharp Monocular Metric Depth in Less Than a Second</a></h2>
+        </div>
+"""
+
 
 def run(input_image_path):
     config = DepthProConfig(
@@ -40,17 +46,33 @@ def run(input_image_path):
     return depth_map, focallength_px.item()
 
 
-demo = gr.Interface(
-    fn=run,
-    inputs=gr.Image(label="Input Image", type="filepath"),
-    outputs=[
-        gr.Image(label="Depth Map"),
-        gr.Number(label="Focal Length"),
-    ],
-    description="""
-        <div align="center">
-            <h2><a href="https://arxiv.org/abs/2410.02073">Depth Pro: Sharp Monocular Metric Depth in Less Than a Second</a></h2>
-        </div>
-        """,
-)
+with gr.Blocks() as demo:
+    gr.Markdown(MARKDOWN)
+    with gr.Row():
+        with gr.Column():
+            input_image_path = gr.Image(
+                label="Input Image", type="filepath", sources=["upload"]
+            )
+        with gr.Column():
+            with gr.Column():
+                output_depth_map = gr.Image(label="Depth Map")
+                output_focal_length = gr.Number(label="Focal Length")
+
+    with gr.Row():
+        btn = gr.Button("Run")
+
+    btn.click(
+        run, inputs=[input_image_path], outputs=[output_depth_map, output_focal_length]
+    )
+
+    examples = gr.Examples(
+        examples=[
+            "assets/input_one.webp",
+        ],
+        fn=run,
+        inputs=[input_image_path],
+        outputs=[output_depth_map, output_focal_length],
+        cache_examples=True,
+    )
+
 demo.launch()