Binlaveloos commited on
Commit
2924302
1 Parent(s): 93d0173

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -9,10 +9,10 @@ import os
9
  import spaces
10
  import json
11
 
12
- from gradio_depth_pred import create_demo as create_depth_pred_demo
13
- from gradio_im_to_3d import create_demo as create_im_to_3d_demo
14
 
15
- model = torch.hub.load('isl-org/ZoeDepth', "ZoeD_N", pretrained=True).to('cuda').eval()
16
 
17
  #dpt_beit = pipeline(task = "depth-estimation", model="Intel/dpt-beit-base-384", device=0)
18
  dpt_beit = pipeline(task = "depth-estimation", model="Intel/dpt-beit-large-512", device=0)
@@ -98,8 +98,8 @@ with gr.Blocks(css=css) as demo:
98
  gr.Markdown("bla bla description")
99
  with gr.Tab("Depth Prediction"):
100
  create_depth_pred_demo(model)
101
- with gr.Tab("Image to 3D"):
102
- create_im_to_3d_demo(model)
103
 
104
  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>
105
  <p><img src="https://visitor-badge.glitch.me/badge?page_id=shariqfarooq.zoedepth_demo_hf" alt="visitors"></p></center>''')
 
9
  import spaces
10
  import json
11
 
12
+ #from gradio_depth_pred import create_demo as create_depth_pred_demo
13
+ #from gradio_im_to_3d import create_demo as create_im_to_3d_demo
14
 
15
+ #model = torch.hub.load('isl-org/ZoeDepth', "ZoeD_N", pretrained=True).to('cuda').eval()
16
 
17
  #dpt_beit = pipeline(task = "depth-estimation", model="Intel/dpt-beit-base-384", device=0)
18
  dpt_beit = pipeline(task = "depth-estimation", model="Intel/dpt-beit-large-512", device=0)
 
98
  gr.Markdown("bla bla description")
99
  with gr.Tab("Depth Prediction"):
100
  create_depth_pred_demo(model)
101
+ #with gr.Tab("Image to 3D"):
102
+ # create_im_to_3d_demo(model)
103
 
104
  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>
105
  <p><img src="https://visitor-badge.glitch.me/badge?page_id=shariqfarooq.zoedepth_demo_hf" alt="visitors"></p></center>''')