Update app.py

app.py

@@ -118,17 +118,32 @@ def predict(img_rgb: np.ndarray):
     return depth_color_rgb, depth_gray, npy_path
 
 # ================== Gradio UI ==================
-demo = gr.Interface(
-    fn=predict,
-    inputs=gr.Image(type="numpy", label="Input Image"),
-    outputs=[
-        gr.Image(type="numpy", label="Depth (Color)"),
-        gr.Image(type="numpy", label="Depth (Gray)"),
-        gr.File(label="Depth (.npy)"),
-    ],
-    title="DAP Depth Prediction Demo",
-    description="Upload an image and get depth prediction (color / gray / npy).",
-)
+example_paths = [
+    "hfdemo/01.jpg",
+    "hfdemo/02.jpg",
+    "hfdemo/03.jpg",
+    "hfdemo/04.jpg",
+    "hfdemo/05.jpg",
+    "hfdemo/06.jpg",
+]
+
+with gr.Blocks() as demo:
+    gr.Markdown("# DAP Depth Prediction Demo\nChoose an example or upload your own image.")
+
+    inp = gr.Image(type="numpy", label="Input Image")
+    out_color = gr.Image(type="numpy", label="Depth (Color)")
+    out_gray = gr.Image(type="numpy", label="Depth (Gray)")
+    out_npy = gr.File(label="Depth (.npy)")
+
+    btn = gr.Button("Run")
+
+    btn.click(fn=predict, inputs=inp, outputs=[out_color, out_gray, out_npy])
+
+    gr.Examples(
+        examples=example_paths,
+        inputs=inp,
+        label="Examples (click to load)",
+    )
 
 demo.launch(
     server_name="0.0.0.0",