Update app.py

app.py

@@ -39,7 +39,7 @@ def GenerateDrums(input_midi, input_num_tokens):
     model = TransformerWrapper(
         num_tokens = PAD_IDX+1,
         max_seq_len = SEQ_LEN,
-        attn_layers = Decoder(dim = 1024, depth = 8, heads = 16, attn_flash = True)
+        attn_layers = Decoder(dim = 1024, depth = 4, heads = 16, attn_flash = True)
         )
     
     model = AutoregressiveWrapper(model, ignore_index = PAD_IDX)