asigalov61 commited on
Commit
106d795
·
verified ·
1 Parent(s): ec158fc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ print('Loading model...')
30
 
31
  SEQ_LEN = 1802
32
  PAD_IDX = 771
33
- DEVICE = 'cuda' # 'cpu'
34
 
35
  # instantiate the model
36
 
 
30
 
31
  SEQ_LEN = 1802
32
  PAD_IDX = 771
33
+ DEVICE = 'cpu' # 'cuda'
34
 
35
  # instantiate the model
36