Hugging Face's logo

Spaces:
gangweix
/
Pixel-Perfect-Depth
Running on Zero

App Files Community
1
gangweix commited on
Commit
778b9d0
·
verified ·
1 Parent(s): 4874a88

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -44,7 +44,8 @@ set_seed(666)
44
  def main(share=True):
45
  print("Initializing Pixel-Perfect Depth Demo...")
46
 
47
- DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
48
  default_steps = 10
49
  model = PixelPerfectDepth(sampling_steps=default_steps)
50
  ckpt_path = hf_hub_download(
 
44
  def main(share=True):
45
  print("Initializing Pixel-Perfect Depth Demo...")
46
 
47
+ # DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
48
+ DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
49
  default_steps = 10
50
  model = PixelPerfectDepth(sampling_steps=default_steps)
51
  ckpt_path = hf_hub_download(