Update app.py

app.py

@@ -117,7 +117,7 @@ def GenerateDrums(input_midi, input_num_tokens):
     output = []
 
     temperature=0.9
-    max_drums_limit=8
+    max_drums_limit=4
     num_memory_tokens=4096
 
     for c in comp_times[:input_num_tokens]:
@@ -128,8 +128,11 @@ def GenerateDrums(input_midi, input_num_tokens):
         o = 128
         
         ncount = 0
+
+        time = 0
+        ntime = output[-1]
         
-        while o > 127 and ncount < max_drums_limit:
+        while o > 127 and ncount < max_drums_limit and time < ntime:
           with ctx:
             out = model.generate(x[-num_memory_tokens:],
                                 1,
@@ -138,11 +141,15 @@ def GenerateDrums(input_midi, input_num_tokens):
                                 verbose=False)
         
           o = out.tolist()[0][0]
+
+          if 128 <= o < 256:
+              time += (o-128)
+              ncount = 0
         
           if 256 <= o < 384:
             ncount += 1
         
-          if o > 127:
+          if o > 127 and time < ntime:
             x = torch.cat((x, out), 1)
         
         x_output = x.tolist()[0][len(output):]
@@ -201,10 +208,12 @@ def GenerateDrums(input_midi, input_num_tokens):
                 pitch = (ss-256)
     
             if 384 <= ss < 393:
+
+                if pitch != 0:
     
-                vel = (ss-384) * 15
-    
-                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',
@@ -248,8 +257,6 @@ def GenerateDrums(input_midi, input_num_tokens):
     print('-' * 70)
     print('Req execution time:', (reqtime.time() - start_time), 'sec')
 
-    print([output_midi_title, output_midi_summary, output_midi, output_audio, output_plot])
-    
     return [output_midi_title, output_midi_summary, output_midi, output_audio, output_plot]
 
 # =================================================================================================