Spaces:

gangweix
/

Pixel-Perfect-Depth

Running on Zero

gangweix commited on Oct 11

Commit

83538b8

verified ·

1 Parent(s): 9e3b2a7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -51,8 +51,8 @@ css = """
 set_seed(666)
-# DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-DEVICE = torch.device("cuda")
 default_steps = 20
 model = PixelPerfectDepth(sampling_steps=default_steps)

 set_seed(666)
+DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+# DEVICE = torch.device("cuda")
 default_steps = 20
 model = PixelPerfectDepth(sampling_steps=default_steps)