Spaces:
Running
on
Zero
Running
on
Zero
lemonaddie
commited on
Update app_recon.py
Browse files- app_recon.py +1 -0
app_recon.py
CHANGED
@@ -54,6 +54,7 @@ import torchvision.transforms.functional as TF
|
|
54 |
from torchvision.transforms import InterpolationMode
|
55 |
|
56 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
57 |
|
58 |
# Initialization.. Downloading the ckpts
|
59 |
|
|
|
54 |
from torchvision.transforms import InterpolationMode
|
55 |
|
56 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
57 |
+
print("Our current device is ", device)
|
58 |
|
59 |
# Initialization.. Downloading the ckpts
|
60 |
|