Spaces:
Sleeping
Sleeping
asigalov61
commited on
Commit
•
38a9fd9
1
Parent(s):
5dda586
Update app.py
Browse files
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 = '
|
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 |
|