Update app.py

app.py

@@ -396,11 +396,14 @@ if __name__ == "__main__":
         
         gr.Markdown("## Select generation options")
 
-        input_freestyle_continuation = gr.Checkbox(label="Freestyle continuation", value=False)
         input_number_prime_chords = gr.Slider(0, 512, value=128, step=8, label="Number of prime chords")
         input_use_original_durations = gr.Checkbox(label="Use original durations", value=False)
         input_match_original_pitches_counts = gr.Checkbox(label="Match original pitches counts", value=False)
 
+        input_freestyle_continuation = gr.Checkbox(label="Freestyle continuation", value=False)
+        input_number_prime_tokens = gr.Slider(0, 1024, value=512, step=16, label="Number of prime tokens")
+        input_number_gen_tokens = gr.Slider(0, 2048, value=512, step=16, label="Number of tokens to generate")
+
         run_btn = gr.Button("generate", variant="primary")
 
         gr.Markdown("## Generation results")
@@ -411,21 +414,25 @@ if __name__ == "__main__":
         output_plot = gr.Plot(label="Output MIDI score plot")
         output_midi = gr.File(label="Output MIDI file", file_types=[".mid"])
 
-        run_event = run_btn.click(Generate_Rock_Song, [input_freestyle_continuation, 
-                                                       input_number_prime_chords, 
+        run_event = run_btn.click(Generate_Rock_Song, [input_number_prime_chords, 
                                                        input_use_original_durations,
-                                                       input_match_original_pitches_counts
+                                                       input_match_original_pitches_counts,
+                                                       input_freestyle_continuation,
+                                                       input_number_prime_tokens,
+                                                       input_number_gen_tokens
                                                       ],
                                   [output_midi_title, output_midi_summary, output_midi, output_audio, output_plot])
 
         gr.Examples(
-            [["Sharing The Night Together.kar", False, 256, True, True],
-             ["Hotel California.mid", False, 256, True, True],
+            [["Sharing The Night Together.kar", 256, True, True, False, 512, 1024],
+             ["Hotel California.mid", 256, True, True, False, 512, 1024],
             ],
-            [input_freestyle_continuation, 
-             input_number_prime_chords, 
+            [input_number_prime_chords, 
              input_use_original_durations,
              input_match_original_pitches_counts,
+             input_freestyle_continuation,
+             input_number_prime_tokens,
+             input_number_gen_tokens
             ],
             [output_midi_title, output_midi_summary, output_midi, output_audio, output_plot],
             Generate_Rock_Song,