Update app.py

app.py

@@ -1,4 +1,4 @@
-# https://huggingface.co/spaces/asigalov61/Melody2Song-Seq2Seq-Music-Transformer
+# https://huggingface.co/spaces/asigalov61/Bridge-Music-Transformer
 
 import os
 import time as reqtime
@@ -31,8 +31,8 @@ def GenerateSong(input_melody_seed_number):
 
     print('Loading model...')
 
-    SEQ_LEN = 2560
-    PAD_IDX = 514
+    SEQ_LEN = 3074
+    PAD_IDX = 653
     DEVICE = 'cuda' # 'cuda'
 
     # instantiate the model
@@ -43,7 +43,7 @@ def GenerateSong(input_melody_seed_number):
         attn_layers = Decoder(dim = 1024, depth = 32, heads = 16, attn_flash = True)
         )
     
-    model = AutoregressiveWrapper(model, ignore_index = PAD_IDX)
+    model = AutoregressiveWrapper(model, ignore_index = PAD_IDX, pad_value=PAD_IDX)
 
     model.to(DEVICE)
     print('=' * 70)
@@ -203,10 +203,10 @@ if __name__ == "__main__":
     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>")
+        gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Bridge Music Transformer</h1>")
+        gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Generate a seamless bridge between two composition parts</h1>")
         gr.Markdown(
-            "![Visitors](https://api.visitorbadge.io/api/visitors?path=asigalov61.Melody2Song-Seq2Seq-Music-Transformer&style=flat)\n\n")
+            "![Visitors](https://api.visitorbadge.io/api/visitors?path=asigalov61.Bridge-Music-Transformer&style=flat)\n\n")
         
         input_melody_seed_number = gr.Slider(0, 203664, value=0, step=1, label="Select seed melody number")