Update app.py

app.py

@@ -9,11 +9,14 @@ import os
 import spaces
 import json
 
-dpt_beit = pipeline(task = "depth-estimation", model="Intel/dpt-beit-base-384", device=0)
-#dpt_beit = pipeline(task = "depth-estimation", model="Intel/dpt-beit-large-512", device=0)
+from gradio_depth_pred import create_demo as create_depth_pred_demo
+from gradio_im_to_3d import create_demo as create_im_to_3d_demo
 
-depth_anything = pipeline(task = "depth-estimation", model="nielsr/depth-anything-small", device=0)
-#depth_anything = pipeline(task = "depth-estimation", model="LiheYoung/depth-anything-large-hf", device=0)
+#dpt_beit = pipeline(task = "depth-estimation", model="Intel/dpt-beit-base-384", device=0)
+dpt_beit = pipeline(task = "depth-estimation", model="Intel/dpt-beit-large-512", device=0)
+
+#depth_anything = pipeline(task = "depth-estimation", model="nielsr/depth-anything-small", device=0)
+depth_anything = pipeline(task = "depth-estimation", model="LiheYoung/depth-anything-large-hf", device=0)
 
 dpt_large = pipeline(task = "depth-estimation", model="intel/dpt-large", device=0)
 
@@ -42,6 +45,19 @@ css = """
     border: 1px solid #ccc; 
   }
 """
+
+css_zoe = """
+#img-display-container {
+    max-height: 50vh;
+    }
+#img-display-input {
+    max-height: 40vh;
+    }
+#img-display-output {
+    max-height: 40vh;
+    }
+    
+"""
 with gr.Blocks(css=css) as demo:
   gr.HTML("<h1><center>Compare Depth Estimation Models<center><h1>")
   gr.Markdown("In this Space, you can compare different depth estimation models: [DPT-Large](https://huggingface.co/Intel/dpt-large), [DPT with BeiT backbone](https://huggingface.co/Intel/dpt-beit-large-512) and the recent [Depth Anything Model small checkpoint](https://huggingface.co/LiheYoung/depth-anything-small-hf). 🤩")
@@ -52,9 +68,9 @@ with gr.Blocks(css=css) as demo:
       with gr.Row():
         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")
+          output_1 = gr.Image(type="pil", label="Intel dpt-large")
+          output_2 = gr.Image(type="pil", label="DPT with BeiT Backbone, dpt-beit-large-512")
+          output_3 = gr.Image(type="pil", label="LiheYoung/depth-anything-large-hf")
 
   gr.Examples([["bee.jpg"], ["cat.png"], ["cats.png"]],
   inputs = input_img,
@@ -66,5 +82,19 @@ with gr.Blocks(css=css) as demo:
 
   input_img.change(infer, [input_img], [output_1, output_2, output_3])
 
+with gr.Blocks(css=css|css_zoe) as demo2:
+    gr.Markdown(title)
+    gr.Markdown(description)
+    with gr.Tab("Depth Prediction"):
+        create_depth_pred_demo(model)
+    with gr.Tab("Image to 3D"):
+        create_im_to_3d_demo(model)
+    with gr.Tab("360 Panorama to 3D"):
+        create_pano_to_3d_demo(model)
+
+    gr.HTML('''<br><br><br><center>You can duplicate this Space to skip the queue:<a href="https://huggingface.co/spaces/shariqfarooq/ZoeDepth?duplicate=true"><img src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a><br>
+        <p><img src="https://visitor-badge.glitch.me/badge?page_id=shariqfarooq.zoedepth_demo_hf" alt="visitors"></p></center>''')
+
+
 
 demo.launch(debug=True, share=True)