RamAnanth1 commited on
Commit
8f3f6b7
1 Parent(s): 926e9ff

Use Zoe_NK model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ description = "Unofficial Gradio Demo for using ZoeDepth: Zero-shot Transfer by
9
  examples = [["example.png"],["example_2.png"]]
10
  repo = "isl-org/ZoeDepth"
11
  # Zoe_N
12
- model_zoe_n = torch.hub.load(repo, "ZoeD_N", pretrained=True)
13
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
14
  zoe = model_zoe_n.to(DEVICE)
15
 
 
9
  examples = [["example.png"],["example_2.png"]]
10
  repo = "isl-org/ZoeDepth"
11
  # Zoe_N
12
+ model_zoe_n = torch.hub.load(repo, "ZoeD_NK", pretrained=True)
13
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
14
  zoe = model_zoe_n.to(DEVICE)
15