Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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)+
|
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)
|