Gregniuki commited on
Commit
6f9ec66
1 Parent(s): b26481f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ def gpu_decorator(func):
32
  return func
33
 
34
  device = (
35
- "cuda"
36
  if torch.cuda.is_available()
37
  else "mps" if torch.backends.mps.is_available() else "cpu"
38
  )
 
32
  return func
33
 
34
  device = (
35
+ "cpu"
36
  if torch.cuda.is_available()
37
  else "mps" if torch.backends.mps.is_available() else "cpu"
38
  )