asigalov61 commited on
Commit
9da1fa1
1 Parent(s): 9dce73b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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