Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -52,7 +52,6 @@ css = """
|
|
| 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)
|
|
|
|
| 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)
|