test

app.py

@@ -29,38 +29,22 @@ def normalize_depth(depth_map):
     normalized_depth = ((depth_map - depth_map.min()) / (depth_map.max() - depth_map.min())) * 255
     return normalized_depth.astype(np.uint8)
 
-# Create a Gradio interface
-iface = gr.Interface(
-    fn=estimate_depths, 
-    inputs=gr.Image(type="pil"), 
-    outputs=[
-        gr.Image(type="numpy", label="LiheYoung/depth-anything-base-hf"),
-        gr.Image(type="numpy", label="LiheYoung/depth-anything-small-hf"),
-        gr.Image(type="numpy", label="Intel/dpt-swinv2-tiny-256"),
-        gr.Image(type="numpy", label="Intel/dpt-beit-base-384")
-    ],
-    title="Multi-Model Depth Estimation",
-    description="Upload an image to get depth estimation maps from multiple models.",
-    layout="vertical"
-)
+# Create a Gradio interface using Blocks
+with gr.Blocks() as iface:
+    gr.Markdown("# Multi-Model Depth Estimation\nUpload an image to get depth estimation maps from multiple models.")
+    
+    with gr.Row():
+        input_image = gr.Image(type="pil", label="Input Image")
+    
+    with gr.Row():
+        with gr.Column():
+            output_base = gr.Image(type="numpy", label="LiheYoung/depth-anything-base-hf", interactive=False)
+            output_small = gr.Image(type="numpy", label="LiheYoung/depth-anything-small-hf", interactive=False)
+        with gr.Column():
+            output_intel = gr.Image(type="numpy", label="Intel/dpt-swinv2-tiny-256", interactive=False)
+            output_beit = gr.Image(type="numpy", label="Intel/dpt-beit-base-384", interactive=False)
+
+    input_image.change(fn=estimate_depths, inputs=input_image, outputs=[output_base, output_small, output_intel, output_beit])
 
 # Launch the Gradio app
 iface.launch()
-
-
-""" 
-from transformers import pipeline
-from PIL import Image
-import requests
-
-# load pipe
-pipe = pipeline(task="depth-estimation", model="LiheYoung/depth-anything-small-hf")
-
-# load image
-url = 'http://images.cocodataset.org/val2017/000000039769.jpg'
-image = Image.open(requests.get(url, stream=True).raw)
-
-# inference
-depth = pipe(image)["depth"]
-
-"""