Update app.py

app.py

@@ -21,7 +21,7 @@ css = """
     
 """
 DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
-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("cpu").eval()
 
 # config_mode="infer"
 # pretrained_resource = f"local::C:/Users/Charl/.cache/torch/hub/checkpoints/ZoeD_M12_N.pt"
@@ -35,9 +35,9 @@ with gr.Blocks(css=css) as API:
     # gr.Markdown(title)
     # gr.Markdown(description)
     with gr.Tab("Depth Prediction"):
-        depth_interface(model)
+        depth_interface(model, DEVICE)
     with gr.Tab("Image to 3D"):
-        mesh_interface(model)
+        mesh_interface(model, DEVICE)
     # with gr.Tab("360 Panorama to 3D"):
     #     create_pano_to_3d_demo(model)