Spaces:
Running
Running
Commit
·
7be757f
1
Parent(s):
112ada6
fix
Browse files
app.py
CHANGED
|
@@ -18,6 +18,7 @@ from models.models import BiFormer
|
|
| 18 |
|
| 19 |
DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 20 |
# DEVICE = torch.device('cpu')
|
|
|
|
| 21 |
|
| 22 |
# Configure logging
|
| 23 |
logging.basicConfig(level=logging.INFO)
|
|
|
|
| 18 |
|
| 19 |
DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 20 |
# DEVICE = torch.device('cpu')
|
| 21 |
+
print(DEVICE)
|
| 22 |
|
| 23 |
# Configure logging
|
| 24 |
logging.basicConfig(level=logging.INFO)
|