Update app.py

app.py

@@ -58,7 +58,7 @@ 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_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'
                     }
 
 SOUDFONT_PATH = 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2'
@@ -135,7 +135,11 @@ def load_midi(input_midi, model_selector=''):
         score.append(max(0, min(127, c[0][1]-pc[0][1])))
     
         for n in c:
-            score.extend([max(1, min(127, n[2]))+128, max(1, min(127, n[4]))+256, max(1, min(127, n[5]))+384])
+            if model_selector == 'with velocity':
+                score.extend([max(1, min(127, n[2]))+128, max(1, min(127, n[4]))+256, max(1, min(127, n[5]))+384])
+
+            else:
+                score.extend([max(1, min(127, n[2]))+128, max(1, min(127, n[4]))+256])
     
         pc = c
 
@@ -168,6 +172,9 @@ def save_midi(tokens, batch_number=None, model_selector=''):
         elif 256 < m < 384:
             pitch = (m-256)
 
+            if model_selector == 'without velocity':
+                song_f.append(['note', time, dur, 0, pitch, max(40, pitch), 0])
+
         elif 384 < m < 512:
             vel = (m-384)