Update app.py

app.py

@@ -31,8 +31,8 @@ def GenerateSong(input_melody_seed_number):
 
     print('Loading model...')
 
-    SEQ_LEN = 2560
-    PAD_IDX = 514
+    SEQ_LEN = 1024
+    PAD_IDX = 14627
     DEVICE = 'cuda' # 'cuda'
 
     # instantiate the model
@@ -40,7 +40,7 @@ def GenerateSong(input_melody_seed_number):
     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 = 12, heads = 16, attn_flash = True)
         )
     
     model = AutoregressiveWrapper(model, ignore_index = PAD_IDX)
@@ -51,7 +51,7 @@ def GenerateSong(input_melody_seed_number):
     print('Loading model checkpoint...')
 
     model.load_state_dict(
-        torch.load('Melody2Song_Seq2Seq_Music_Transformer_Trained_Model_28482_steps_0.719_loss_0.7865_acc.pth',
+        torch.load('Annotated_MIDI_Dataset_Classifier_Trained_Model_21269_steps_0.4335_loss_0.8716_acc.pth',
                    map_location=DEVICE))
     print('=' * 70)
 
@@ -201,7 +201,7 @@ if __name__ == "__main__":
     soundfont = "SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2"
 
     print('Loading seed meldoies data...')
-    seed_melodies_data = TMIDIX.Tegridy_Any_Pickle_File_Reader('Melody2Song_Seq2Seq_Music_Transformer_Seed_Melodies_Data')
+    seed_melodies_data = TMIDIX.Tegridy_Any_Pickle_File_Reader('processed_scores')
     print('=' * 70)
    
     app = gr.Blocks()