Spaces:
hysts
/
Running on Zero

osanseviero HF staff commited on
Commit
b5308e1
1 Parent(s): 2e7be44

Demo works with CPU

Browse files

The demo says that it only works with GPU, but it seems to be working well in CPU

Files changed (1) hide show
  1. app.py +1 -1
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 🥶 This demo does not work 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