Spaces:
Configuration error
Configuration error
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 |
|