Update app.py

app.py

@@ -50,7 +50,7 @@ def GenerateDrums(input_midi, input_num_tokens):
     print('Loading model checkpoint...')
 
     model.load_state_dict(
-        torch.load('Ultimate_Drums_Transformer_Small_Trained_Model_VER3_VEL_11222_steps_0.5749_loss_0.8085_acc.pth',
+        torch.load('Ultimate_Drums_Transformer_Small_Trained_Model_VER4_RST_VEL_4L_16534_steps_0.4074_loss_0.8631_acc.pth',
                    map_location=DEVICE))
     print('=' * 70)
 
@@ -168,6 +168,7 @@ def GenerateDrums(input_midi, input_num_tokens):
         time = 0
         dtime = 0
         ntime = 0
+        ptime = 0
         dur = 32
         vel = 90
         vels = [100, 120]
@@ -180,6 +181,8 @@ def GenerateDrums(input_midi, input_num_tokens):
     
             if 0 <= ss < 128:
 
+                dtime = ptime = time
+
                 time += cscore[idx][0][0] * 32
     
                 for c in cscore[idx]:
@@ -198,13 +201,10 @@ def GenerateDrums(input_midi, input_num_tokens):
                 pitch = (ss-256)
     
             if 384 <= ss < 393:
-            
-                vel = ((ss-384)+1) * 15
-            
-                if dtime == time:
-                    song_f.append(['note', time, dur, 9, pitch, vel, 128])
-                else:
-                    song_f.append(['note', dtime, dur, 9, pitch, vel, 128])
+    
+                vel = (ss-384) * 15
+    
+                song_f.append(['note', dtime, dur, 9, pitch, vel, 128])
     
     detailed_stats = TMIDIX.Tegridy_ms_SONG_to_MIDI_Converter(song_f,
                                                               output_signature = 'Ultimate Drums Transformer',