Hugging Face's logo

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

App Files Community
1
gangweix commited on
Commit
ce2b252
·
verified ·
1 Parent(s): 2a11767

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -51,7 +51,8 @@ css = """
51
 
52
  set_seed(666)
53
 
54
- DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
55
 
56
  default_steps = 20
57
  model = PixelPerfectDepth(sampling_steps=default_steps)
 
51
 
52
  set_seed(666)
53
 
54
+ # DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
55
+ DEVICE = torch.device("cuda")
56
 
57
  default_steps = 20
58
  model = PixelPerfectDepth(sampling_steps=default_steps)