Update app.py

app.py

@@ -120,7 +120,7 @@ def GenerateDrums(input_midi, input_num_tokens, input_top_k_value):
     output = []
 
     temperature=0.9
-    max_drums_limit=3
+    max_drums_limit=5
     num_memory_tokens=4096
 
     for c in comp_times[:input_num_tokens]:
@@ -290,7 +290,7 @@ if __name__ == "__main__":
         
         input_midi = gr.File(label="Input MIDI", file_types=[".midi", ".mid", ".kar"])
         input_num_tokens = gr.Slider(16, 2048, value=256, step=16, label="Number of composition chords to generate drums for")
-        input_top_k_value = gr.Slider(1, 50, value=1, step=1, label="Model sampling top_k value")
+        input_top_k_value = gr.Slider(1, 50, value=5, step=1, label="Model sampling top_k value")
         
         run_btn = gr.Button("generate", variant="primary")
 
@@ -307,16 +307,16 @@ if __name__ == "__main__":
                                   [output_midi_title, output_midi_summary, output_midi, output_audio, output_plot])
 
         gr.Examples(
-            [["Ultimate-Drums-Transformer-Melody-Seed-1.mid", 128, 1], 
-             ["Ultimate-Drums-Transformer-Melody-Seed-2.mid", 128, 1], 
-             ["Ultimate-Drums-Transformer-Melody-Seed-3.mid", 128, 1],
-             ["Ultimate-Drums-Transformer-Melody-Seed-4.mid", 128, 1],
-             ["Ultimate-Drums-Transformer-Melody-Seed-5.mid", 128, 1],
-             ["Ultimate-Drums-Transformer-Melody-Seed-6.mid", 128, 1],
-             ["Ultimate-Drums-Transformer-MI-Seed-1.mid", 128, 1],
-             ["Ultimate-Drums-Transformer-MI-Seed-2.mid", 128, 1],
-             ["Ultimate-Drums-Transformer-MI-Seed-3.mid", 128, 1],
-             ["Ultimate-Drums-Transformer-MI-Seed-4.mid", 128, 1]],
+            [["Ultimate-Drums-Transformer-Melody-Seed-1.mid", 128, 5], 
+             ["Ultimate-Drums-Transformer-Melody-Seed-2.mid", 128, 5], 
+             ["Ultimate-Drums-Transformer-Melody-Seed-3.mid", 128, 5],
+             ["Ultimate-Drums-Transformer-Melody-Seed-4.mid", 128, 5],
+             ["Ultimate-Drums-Transformer-Melody-Seed-5.mid", 128, 5],
+             ["Ultimate-Drums-Transformer-Melody-Seed-6.mid", 128, 5],
+             ["Ultimate-Drums-Transformer-MI-Seed-1.mid", 128, 5],
+             ["Ultimate-Drums-Transformer-MI-Seed-2.mid", 128, 5],
+             ["Ultimate-Drums-Transformer-MI-Seed-3.mid", 128, 5],
+             ["Ultimate-Drums-Transformer-MI-Seed-4.mid", 128, 5]],
             [input_midi, input_num_tokens, input_top_k_value],
             [output_midi_title, output_midi_summary, output_midi, output_audio, output_plot],
             GenerateDrums,