asigalov61 commited on
Commit
61daf8d
1 Parent(s): eec6956

Update app.py

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