Update app.py

app.py

@@ -13,19 +13,19 @@ dpt_beit = pipeline(task = "depth-estimation", model="Intel/dpt-beit-base-384")
 depth_anything = pipeline(task = "depth-estimation", model="nielsr/depth-anything-small")
 dpt_large = pipeline(task = "depth-estimation", model="intel/dpt-large")
 @spaces.GPU
-def depth_anything_inference(image_path):
-  return depth_anything(image_path)["depth"]
+def depth_anything_inference(img):
+  return depth_anything(img)["depth"]
 @spaces.GPU
-def dpt_beit_inference(image):
-  return dpt_beit(image)["depth"]
+def dpt_beit_inference(img):
+  return dpt_beit(img)["depth"]
 
 @spaces.GPU
-def dpt_large_inference(image):
-  return dpt_large(image)["depth"]
+def dpt_large_inference(img):
+  return dpt_large(img)["depth"]
     
 
-def infer(image):
-  return dpt_large_inference(image), dpt_beit_inference(image), depth_anything_inference(image)
+def infer(img):
+  return dpt_large_inference(img), dpt_beit_inference(img), depth_anything_inference(img)
 
 
 css = """
@@ -39,12 +39,19 @@ with gr.Blocks(css=css) as demo:
   gr.HTML("<h1><center>Compare Depth Estimation Models<center><h1>")
   with gr.Column():
       with gr.Row():
-        input_img = gr.Image(label="Input Image")
+        input_img = gr.Image(label="Input Image", type="pil")
       with gr.Row():
           output_1 = gr.Image(type="pil", label="DPT-Large")
           output_2 = gr.Image(type="pil", label="DPT with BeiT Backbone")
           output_3 = gr.Image(type="pil", label="Depth Anything")
-  
+
+  gr.Examples([["bee.jpg"]],
+  inputs = input_img,
+  outputs = [output_1, output_2, output_3],
+  fn=infer,
+  cache_examples=True,
+  label='Click on any Examples below to get depth estimation results quickly 👇'
+  )
 
   input_img.change(infer, [input_img], [output_1, output_2, output_3])