Update app.py

app.py

@@ -58,8 +58,8 @@ print('=' * 70)
 
 MODEL_CHECKPOINTS = {
                     'with velocity - 3 epochs': 'Monster_Piano_Transformer_Velocity_Trained_Model_59896_steps_0.9055_loss_0.735_acc.pth',
-                    'without velocity - 3 epochs': 'Monster_Piano_Transformer_No_Velocity_Trained_Model_50647_steps_0.8166_loss_0.7561_acc.pth',
-                    'without velocity - 5 epochs': 'Monster_Piano_Transformer_No_Velocity_Trained_Model_84419_steps_0.7474_loss_0.7782_acc.pth'
+                    'without velocity - 3 epochs': 'Monster_Piano_Transformer_No_Velocity_Trained_Model_69412_steps_0.8577_loss_0.7442_acc.pth',
+                    'without velocity - 7 epochs': 'Monster_Piano_Transformer_No_Velocity_Trained_Model_161960_steps_0.7775_loss_0.7661_acc.pth'
                     }
 
 SOUDFONT_PATH = 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2'
@@ -82,7 +82,12 @@ def load_model(model_selector):
     ctx = torch.amp.autocast(device_type=device_type, dtype=ptdtype)
     
     SEQ_LEN = 2048
-    PAD_IDX = 512
+
+    if model_selector == 'with velocity - 3 epochs':
+        PAD_IDX = 512
+
+    else:
+        PAD_IDX = 384
     
     model = TransformerWrapper(
             num_tokens = PAD_IDX+1,
@@ -173,7 +178,7 @@ def save_midi(tokens, batch_number=None, model_selector=''):
         elif 256 < m < 384:
             pitch = (m-256)
 
-            if model_selector == 'without velocity - 3 epochs' or model_selector == 'without velocity - 5 epochs':
+            if model_selector == 'without velocity - 3 epochs' or model_selector == 'without velocity - 7 epochs':
                 song_f.append(['note', time, dur, 0, pitch, max(40, pitch), 0])
 
         elif 384 < m < 512:
@@ -529,7 +534,7 @@ with gr.Blocks() as demo:
     
     gr.Markdown("## Generate")
 
-    model_selector = gr.Dropdown(["without velocity - 5 epochs",
+    model_selector = gr.Dropdown(["without velocity - 7 epochs",
                                   "without velocity - 3 epochs", 
                                   "with velocity - 3 epochs"
                                 ],