asigalov61 commited on
Commit
38a9fd9
1 Parent(s): 5dda586

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ def GenerateBridge(input_midi, input_start_note):
44
 
45
  SEQ_LEN = 3074
46
  PAD_IDX = 653
47
- DEVICE = 'cuda' # 'cuda'
48
 
49
  # instantiate the model
50
 
 
44
 
45
  SEQ_LEN = 3074
46
  PAD_IDX = 653
47
+ DEVICE = 'cpu' # 'cuda'
48
 
49
  # instantiate the model
50