Update app.py

app.py

@@ -12,6 +12,8 @@ import json
 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
 
+model = torch.hub.load('isl-org/ZoeDepth', "ZoeD_N", pretrained=True).to(DEVICE).eval()
+
 #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)
 
@@ -43,6 +45,15 @@ css = """
     height: 500px; 
     overflow: auto; 
     border: 1px solid #ccc; 
+  #img-display-container {
+    max-height: 50vh;
+    }
+  #img-display-input {
+    max-height: 40vh;
+    }
+  #img-display-output {
+    max-height: 40vh;
+    }
   }
 """
 
@@ -82,15 +93,13 @@ 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:
+with gr.Blocks(css=css) as demo:
     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>''')