Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
b5308e1
1
Parent(s):
2e7be44
Demo works with CPU
Browse filesThe demo says that it only works with GPU, but it seems to be working well in CPU
app.py
CHANGED
@@ -18,7 +18,7 @@ if (SPACE_ID := os.getenv('SPACE_ID')) is not None:
|
|
18 |
if torch.cuda.is_available():
|
19 |
DESCRIPTION += '\n<p>Running on GPU 🔥</p>'
|
20 |
else:
|
21 |
-
DESCRIPTION += '\n<p>Running on CPU 🥶
|
22 |
|
23 |
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
|
24 |
|
18 |
if torch.cuda.is_available():
|
19 |
DESCRIPTION += '\n<p>Running on GPU 🔥</p>'
|
20 |
else:
|
21 |
+
DESCRIPTION += '\n<p>Running on CPU 🥶'
|
22 |
|
23 |
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
|
24 |
|