Update app.py

app.py

@@ -1,6 +1,6 @@
-#============================================================================================
+#==========================================================================
 # https://huggingface.co/spaces/projectlosangeles/Orpheus-Pitches-Inpainter
-#============================================================================================
+#==========================================================================
 
 print('=' * 70)
 print('Orpheus Pitches Inpainter Gradio App')
@@ -56,9 +56,8 @@ print('=' * 70)
 
 #==================================================================================
 
-MODEL_CHECKPOINT = 'Orpheus_Bridge_Music_Transformer_Trained_Model_19571_steps_0.9396_loss_0.7365_acc.pth'
-
-SOUDFONT_PATH = 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2'
+MODEL_CHECKPOINT = 'Orpheus_Music_Transformer_Trained_Model_128497_steps_0.6934_loss_0.7927_acc.pth'
+SOUNDFONT_PATH = 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2'
 
 #==================================================================================
 
@@ -71,18 +70,19 @@ dtype = 'bfloat16'
 ptdtype = {'bfloat16': torch.bfloat16, 'float16': torch.float16}[dtype]
 ctx = torch.amp.autocast(device_type=device_type, dtype=ptdtype)
 
-SEQ_LEN = 1668
+SEQ_LEN = 8192
 PAD_IDX = 18819
 
-model = TransformerWrapper(num_tokens = PAD_IDX+1,
-                           max_seq_len = SEQ_LEN,
-                           attn_layers = Decoder(dim = 2048,
-                                                 depth = 8,
-                                                 heads = 32,
-                                                 rotary_pos_emb = True,
-                                                 attn_flash = True
-                                                 )
-                           )
+model = TransformerWrapper(num_tokens=PAD_IDX + 1,
+                           max_seq_len=SEQ_LEN,
+                           attn_layers=Decoder(
+                               dim=2048,
+                               depth=8,
+                               heads=32,
+                               rotary_pos_emb=True,
+                               attn_flash=True
+                          )
+)
 
 model = AutoregressiveWrapper(model, ignore_index=PAD_IDX, pad_value=PAD_IDX)
 
@@ -329,7 +329,7 @@ def Generate_Music_Bridge(input_midi,
                     
             
             audio = midi_to_colab_audio(new_fn, 
-                                soundfont_path=SOUDFONT_PATH,
+                                soundfont_path=SOUNDFONT_PATH,
                                 sample_rate=16000,
                                 volume_scale=10,
                                 output_for_gradio=True