Update app.py

app.py

@@ -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 = 32, 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('Bridge_Music_Transformer_Trained_Model_30023_steps_0.482_loss_0.8523_acc.pth',
                    map_location=DEVICE))
     print('=' * 70)
 
@@ -199,12 +199,9 @@ if __name__ == "__main__":
     print('=' * 70)
 
     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')
-    print('=' * 70)
    
     app = gr.Blocks()
+    
     with app:
         gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Melody2Song Seq2Seq Music Transformer</h1>")
         gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Generate unique songs from melodies with seq2seq music transformer</h1>")