Hugging Face's logo

Spaces:
asigalov61
/
Popular-Hook-Transformer
Running on Zero

App Files Community
asigalov61 commited on
Commit
ed37c31
•
1 Parent(s): 2d2f60f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ print('=' * 70)
37
 
38
  SEQ_LEN = 512
39
  PAD_IDX = 918
40
- DEVICE = 'cpu'
41
 
42
  #====================================================================================
43
 
 
37
 
38
  SEQ_LEN = 512
39
  PAD_IDX = 918
40
+ DEVICE = 'cuda'
41
 
42
  #====================================================================================
43