Update app.py

app.py

@@ -30,16 +30,16 @@ def Generate_Rock_Song(input_midi, input_melody_seed_number):
 
     print('Loading model...')
 
-    SEQ_LEN = 2560
-    PAD_IDX = 514
-    DEVICE = 'cuda' # 'cuda'
+    SEQ_LEN = 4096
+    PAD_IDX = 673
+    DEVICE = 'cuda' # 'cpu'
 
     # instantiate the model
 
     model = TransformerWrapper(
         num_tokens = PAD_IDX+1,
         max_seq_len = SEQ_LEN,
-        attn_layers = Decoder(dim = 1024, depth = 24, heads = 16, attn_flash = True)
+        attn_layers = Decoder(dim = 1024, depth = 16, heads = 16, rotary_pos_emb=True, attn_flash = True)
         )
     
     model = AutoregressiveWrapper(model, ignore_index = PAD_IDX)
@@ -70,6 +70,11 @@ def Generate_Rock_Song(input_midi, input_melody_seed_number):
 
     fn = os.path.basename(input_midi)
     fn1 = fn.split('.')[0]
+
+    print('=' * 70)
+    print('Requested settings:')
+    print('=' * 70)
+    print('Input MIDI file name:', fn)
     
     #===============================================================================
     # Raw single-track ms score