Spaces:

CharlieAmalet
/

Tools3ox_Depth_Generator

Running on Zero

CharlieAmalet commited on 4 days ago

Commit

07dbe58

verified ·

1 Parent(s): 212d34e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ img {
 """
 # DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
-MODEL = torch.hub.load('isl-org/ZoeDepth', "ZoeD_N", pretrained=True)#.to(DEVICE).eval()
 # ----------- Depth functions
 def save_raw_16bit(depth, fpath="raw.png"):

 """
 # DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
+MODEL = torch.hub.load('isl-org/ZoeDepth', "ZoeD_N", pretrained=True).eval()
 # ----------- Depth functions
 def save_raw_16bit(depth, fpath="raw.png"):