Update app.py

app.py

@@ -307,6 +307,8 @@ def Generate_Chords(input_midi, input_num_prime_chords, input_num_gen_chords):
             with ctx:
                 out = model.generate(x,
                                      1,
+                                     filter_logits_fn=top_p,
+                                     filter_kwargs={'thres': 0.96},
                                      temperature=0.9,
                                      return_prime=False,
                                      verbose=False)
@@ -449,7 +451,7 @@ if __name__ == "__main__":
         gr.Markdown("## Upload your MIDI or select a sample example MIDI")
         
         input_midi = gr.File(label="Input MIDI", file_types=[".midi", ".mid", ".kar"])
-        input_num_prime_chords = gr.Slider(1, 128, value=1, step=1, label="Number of prime chords")
+        input_num_prime_chords = gr.Slider(1, 128, value=32, step=1, label="Number of prime chords")
         input_num_gen_chords = gr.Slider(4, 256, value=128, step=1, label="Number of composition chords to generate progression for")
         
         run_btn = gr.Button("Generate Chords", variant="primary")
@@ -466,13 +468,12 @@ if __name__ == "__main__":
                                   [output_midi_title, output_midi_summary, output_midi, output_audio, output_plot])
 
         gr.Examples(
-            [["Chords-Progressions-Transformer-Piano-Seed-2.mid", 1, 128], 
-             ["Chords-Progressions-Transformer-MI-Seed-1.mid", 16, 128],
-             ["Chords-Progressions-Transformer-MI-Seed-2.mid", 1, 128],
-             ["Chords-Progressions-Transformer-MI-Seed-3.mid", 16, 128],
-             ["Chords-Progressions-Transformer-MI-Seed-4.mid", 1, 128],
-             ["Chords-Progressions-Transformer-MI-Seed-5.mid", 16, 128],
-             ["Chords-Progressions-Transformer-MI-Seed-6.mid", 1, 128]
+            [["Chords-Progressions-Transformer-MI-Seed-1.mid", 32, 128],
+             ["Chords-Progressions-Transformer-MI-Seed-2.mid", 32, 128],
+             ["Chords-Progressions-Transformer-MI-Seed-3.mid", 32, 128],
+             ["Chords-Progressions-Transformer-MI-Seed-4.mid", 32, 128],
+             ["Chords-Progressions-Transformer-MI-Seed-5.mid", 32, 128],
+             ["Chords-Progressions-Transformer-MI-Seed-6.mid", 32, 128]
             ],
             [input_midi, input_num_prime_chords, input_num_gen_chords],
             [output_midi_title, output_midi_summary, output_midi, output_audio, output_plot],