ThunderVVV commited on
Commit
49be5d4
·
1 Parent(s): 0675aa2
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -6,6 +6,9 @@ import torch
6
  print(torch.__version__)
7
  print(torch.version.cuda)
8
 
 
 
 
9
  # download model
10
  print("Downloading model weights")
11
  os.system('wget -q https://huggingface.co/ThunderVVV/HaWoR/resolve/main/external/metric_depth_vit_large_800k.pth -P ./thirdparty/Metric3D/weights/')
 
6
  print(torch.__version__)
7
  print(torch.version.cuda)
8
 
9
+ os.system('export DISPLAY=:0.0')
10
+ os.system('Xvfb :0 -screen 0 640x480x24 &')
11
+
12
  # download model
13
  print("Downloading model weights")
14
  os.system('wget -q https://huggingface.co/ThunderVVV/HaWoR/resolve/main/external/metric_depth_vit_large_800k.pth -P ./thirdparty/Metric3D/weights/')