Update app.py

app.py

@@ -225,10 +225,9 @@ def Generate_Drums(input_midi,
 
             def gen_drums(seq):
 
-                y = 256
-                seq.append((128*128)+38+256) # Drum pitch
+                y = 16640
 
-                while y > 255:
+                while y > 16640:
                 
                     x = torch.LongTensor(seq).to(device_type)
                     
@@ -260,6 +259,9 @@ def Generate_Drums(input_midi,
 
                 final_song.extend(chords[i])
 
+                if i == 0:
+                    final_song.append((128*128)+38+256) # Drum pitch
+
                 final_song = gen_drums(final_song)
 
             #==================================================================