Gregniuki commited on
Commit
235c801
·
verified ·
1 Parent(s): cad4507

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ def gpu_decorator(func):
39
  return func
40
 
41
  device = (
42
- "gpu"
43
  if torch.cuda.is_available()
44
  else "mps" if torch.backends.mps.is_available() else "cpu"
45
  )
 
39
  return func
40
 
41
  device = (
42
+ "cuda"
43
  if torch.cuda.is_available()
44
  else "mps" if torch.backends.mps.is_available() else "cpu"
45
  )