Hugging Face's logo

Spaces:
kadabengaran
/
useful-review-classification
Runtime error

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

update

Browse files
Files changed (1) hide show
  1. app/main.py +2 -3
app/main.py CHANGED
@@ -24,8 +24,7 @@ position: fixed;
24
  left: 0;
25
  bottom: 0;
26
  width: 100%;
27
- background-color: white;
28
- color: #fff;
29
  text-align: center;
30
  }
31
  </style>
@@ -44,7 +43,7 @@ def get_device():
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
 
24
  left: 0;
25
  bottom: 0;
26
  width: 100%;
27
+ color: #e7e7e7;
 
28
  text-align: center;
29
  }
30
  </style>
 
43
  else:
44
  return torch.device('cpu')
45
 
46
+ USE_CUDA = False
47
  device = get_device()
48
  if device.type == 'cuda':
49
  USE_CUDA = True