Update app.py

app.py

@@ -57,8 +57,8 @@ print('=' * 70)
 #==================================================================================
 
 MODEL_CHECKPOINTS = {
-                    'with velocity': 'Monster_Piano_Transformer_Velocity_Trained_Model_59896_steps_0.9055_loss_0.735_acc.pth',
-                    'without velocity': 'Monster_Piano_Transformer_No_Velocity_Trained_Model_50647_steps_0.8166_loss_0.7561_acc.pth'
+                    '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'
                     }
 
 SOUDFONT_PATH = 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2'
@@ -135,7 +135,7 @@ def load_midi(input_midi, model_selector=''):
         score.append(max(0, min(127, c[0][1]-pc[0][1])))
     
         for n in c:
-            if model_selector == 'with velocity':
+            if model_selector == 'with velocity - 3 epochs':
                 score.extend([max(1, min(127, n[2]))+128, max(1, min(127, n[4]))+256, max(1, min(127, n[5]))+384])
 
             else:
@@ -172,13 +172,14 @@ def save_midi(tokens, batch_number=None, model_selector=''):
         elif 256 < m < 384:
             pitch = (m-256)
 
-            if model_selector == 'without velocity':
+            if model_selector == 'without velocity - 3 epochs' or model_selector == 'without velocity - 5 epochs':
                 song_f.append(['note', time, dur, 0, pitch, max(40, pitch), 0])
 
         elif 384 < m < 512:
             vel = (m-384)
 
-            song_f.append(['note', time, dur, 0, pitch, vel, 0])
+            if model_selector == 'with velocity - 3 epochs'
+                song_f.append(['note', time, dur, 0, pitch, vel, 0])
 
     if batch_number == None:
         fname = 'Monster-Piano-Transformer-Music-Composition'
@@ -513,11 +514,10 @@ with gr.Blocks() as demo:
     
     gr.Markdown("## Generate")
 
-    model_selector = gr.Dropdown(["with velocity", 
-                                  "without velocity"
+    model_selector = gr.Dropdown(["without velocity - 3 epochs", 
+                                  "with velocity - 3 epochs"
                                 ], 
                                 label="Select model", 
-                                info="Select desired Monster Piano Transformer model"
                                )
     
     num_prime_tokens = gr.Slider(15, 1024, value=1024, step=1, label="Number of prime tokens")