Hugging Face's logo

Spaces:
kadabengaran
/
useful-review-classification
Runtime error

App Files Community
kadabengaran commited on
Commit
3875a44
·
1 Parent(s): 6410177

update

Browse files
Files changed (1) hide show
  1. app/main.py +1 -1
app/main.py CHANGED
@@ -44,7 +44,7 @@ def get_device():
44
  else:
45
  return torch.device('cpu')
46
 
47
- USE_CUDA = False
48
  device = get_device()
49
  if device.type == 'cuda':
50
  USE_CUDA = True
 
44
  else:
45
  return torch.device('cpu')
46
 
47
+ USE_CUDA = True
48
  device = get_device()
49
  if device.type == 'cuda':
50
  USE_CUDA = True