Update app.py

app.py

@@ -327,9 +327,8 @@ def generate_music_and_state(input_midi, num_prime_tokens, num_gen_tokens, num_m
         final_composition.append(18817) # Outro token
 
     if add_drums:
-        pass
-        
-    
+        final_composition.extend([(128*128)+36+256]) # Drum token (36)
+
     batched_gen_tokens = generate_music(final_composition, num_gen_tokens, num_mem_tokens,
                                         NUM_OUT_BATCHES, model_temperature)