Update app.py

app.py

@@ -190,7 +190,7 @@ if __name__ == "__main__":
     model = TransformerWrapper(
         num_tokens=3088,
         max_seq_len=SEQ_LEN,
-        attn_layers=Decoder(dim=1024, depth=32, heads=8)
+        attn_layers=Decoder(dim=1024, depth=16, heads=8)
     )
 
     model = AutoregressiveWrapper(model)
@@ -203,7 +203,7 @@ if __name__ == "__main__":
     print('Loading model checkpoint...')
 
     model.load_state_dict(
-        torch.load('Allegro_Music_Transformer_Small_Trained_Model_56000_steps_0.9399_loss_0.7374_acc.pth',
+        torch.load('Allegro_Music_Transformer_Tiny_Trained_Model_80000_steps_0.9457_loss_0.7443_acc.pth',
                    map_location='cpu'))
     print('=' * 70)