asigalov61
commited on
Commit
•
9da1fa1
1
Parent(s):
9dce73b
Update app.py
Browse files
app.py
CHANGED
@@ -145,7 +145,7 @@ def GenerateDrums(input_midi, input_num_tokens):
|
|
145 |
if o > 127:
|
146 |
x = torch.cat((x, out), 1)
|
147 |
|
148 |
-
x_output = x.tolist()[0]
|
149 |
|
150 |
output.extend(x_output)
|
151 |
|
|
|
145 |
if o > 127:
|
146 |
x = torch.cat((x, out), 1)
|
147 |
|
148 |
+
x_output = x.tolist()[0][len(output):]
|
149 |
|
150 |
output.extend(x_output)
|
151 |
|