Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
1f827e2
1
Parent(s):
fcd32ca
feat: enforce CPU.
Browse files
app.py
CHANGED
|
@@ -86,7 +86,8 @@ def run_3d_mood(fx, fy, cx, cy, image):
|
|
| 86 |
|
| 87 |
gc.collect()
|
| 88 |
|
| 89 |
-
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
|
| 90 |
|
| 91 |
# Data
|
| 92 |
images = image.astype(np.float32)[None, ...]
|
|
|
|
| 86 |
|
| 87 |
gc.collect()
|
| 88 |
|
| 89 |
+
# device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 90 |
+
device = "cpu"
|
| 91 |
|
| 92 |
# Data
|
| 93 |
images = image.astype(np.float32)[None, ...]
|