asigalov61 commited on
Commit
e3193ec
·
verified ·
1 Parent(s): a863118

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -327,7 +327,7 @@ def generate_music_and_state(input_midi, num_prime_tokens, num_gen_tokens, num_m
327
  final_composition.append(18817) # Outro token
328
 
329
  if add_drums:
330
- final_composition.extend([(128*128)+36+256]) # Drum token (36)
331
 
332
  batched_gen_tokens = generate_music(final_composition, num_gen_tokens, num_mem_tokens,
333
  NUM_OUT_BATCHES, model_temperature)
 
327
  final_composition.append(18817) # Outro token
328
 
329
  if add_drums:
330
+ final_composition.extend([(128*128)+38+256]) # Drum token (38)
331
 
332
  batched_gen_tokens = generate_music(final_composition, num_gen_tokens, num_mem_tokens,
333
  NUM_OUT_BATCHES, model_temperature)