freealise commited on
Commit
3057121
1 Parent(s): 45abddb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -22,6 +22,8 @@ import mediapy
22
  #from depth_anything.dpt import DepthAnything
23
  #from depth_anything.util.transform import Resize, NormalizeImage, PrepareForNet
24
  from huggingface_hub import hf_hub_download
 
 
25
  from depth_anything_v2.dpt import DepthAnythingV2
26
 
27
  DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
 
22
  #from depth_anything.dpt import DepthAnything
23
  #from depth_anything.util.transform import Resize, NormalizeImage, PrepareForNet
24
  from huggingface_hub import hf_hub_download
25
+ from huggingface_hub import snapshot_download
26
+ snapshot_download(repo_id="depth-anything/Depth-Anything-V2", local_dir="./", allow_patterns=["*.py"], ignore_patterns=["app.py"])
27
  from depth_anything_v2.dpt import DepthAnythingV2
28
 
29
  DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'