Spaces:
Sleeping
Sleeping
Commit
·
e330fbf
1
Parent(s):
5eda405
update
Browse files
app.py
CHANGED
|
@@ -11,6 +11,7 @@ import cv2
|
|
| 11 |
import numpy as np
|
| 12 |
import os
|
| 13 |
import torch
|
|
|
|
| 14 |
print(f"Is CUDA available: {torch.cuda.is_available()}")
|
| 15 |
if torch.cuda.is_available():
|
| 16 |
device = "cuda"
|
|
|
|
| 11 |
import numpy as np
|
| 12 |
import os
|
| 13 |
import torch
|
| 14 |
+
|
| 15 |
print(f"Is CUDA available: {torch.cuda.is_available()}")
|
| 16 |
if torch.cuda.is_available():
|
| 17 |
device = "cuda"
|