Update app.py

app.py

@@ -12,7 +12,7 @@ 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()
+model = torch.hub.load('isl-org/ZoeDepth', "ZoeD_N", pretrained=True).to('cuda').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)