Update app.py

app.py

@@ -88,7 +88,7 @@ def GenerateMIDI(num_tok, idrums, iinstr):
     model = TransformerWrapper(
         num_tokens=3088,
         max_seq_len=SEQ_LEN,
-        attn_layers=Decoder(dim=1024, depth=16, heads=8, attn_flash=True)
+        attn_layers=Decoder(dim=1024, depth=32, heads=8, attn_flash=True)
     )
 
     model = AutoregressiveWrapper(model)
@@ -101,7 +101,7 @@ def GenerateMIDI(num_tok, idrums, iinstr):
     print('Loading model checkpoint...')
 
     model.load_state_dict(
-        torch.load('Allegro_Music_Transformer_Tiny_Trained_Model_80000_steps_0.9457_loss_0.7443_acc.pth',
+        torch.load('Allegro_Music_Transformer_Small_Trained_Model_56000_steps_0.9399_loss_0.7374_acc.pth',
                    map_location='cuda'))
     print('=' * 70)