Akash Raj commited on
Commit
f531789
·
1 Parent(s): 0303fd6
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -23,10 +23,10 @@ def normalize_depth(depth_map):
23
  return normalized_depth.astype(np.uint8)
24
 
25
  # Create Gradio output components for each pipeline
26
- output_base = gr.outputs.Image(type="numpy", label="LiheYoung/depth-anything-base-hf")
27
- output_small = gr.outputs.Image(type="numpy", label="LiheYoung/depth-anything-small-hf")
28
- output_intel = gr.outputs.Image(type="numpy", label="Intel/dpt-swinv2-tiny-256")
29
- output_beit = gr.outputs.Image(type="numpy", label="Intel/dpt-beit-base-384")
30
 
31
  # Create Gradio interfaces for each pipeline
32
  iface_base = gr.Interface(process_and_display(pipe_base, output_base), inputs=gr.inputs.Image(type="pil"), outputs=output_base, title="Depth Estimation - LiheYoung/depth-anything-base-hf")
 
23
  return normalized_depth.astype(np.uint8)
24
 
25
  # Create Gradio output components for each pipeline
26
+ output_base = gr.Image(type="numpy", label="LiheYoung/depth-anything-base-hf")
27
+ output_small = gr.Image(type="numpy", label="LiheYoung/depth-anything-small-hf")
28
+ output_intel = gr.Image(type="numpy", label="Intel/dpt-swinv2-tiny-256")
29
+ output_beit = gr.Image(type="numpy", label="Intel/dpt-beit-base-384")
30
 
31
  # Create Gradio interfaces for each pipeline
32
  iface_base = gr.Interface(process_and_display(pipe_base, output_base), inputs=gr.inputs.Image(type="pil"), outputs=output_base, title="Depth Estimation - LiheYoung/depth-anything-base-hf")