Update app.py

app.py

@@ -25,7 +25,8 @@ from midi_to_colab_audio import midi_to_colab_audio
 @spaces.GPU
 def Generate_Rock_Song(input_midi,
                        input_gen_type,
-                       input_number_prime_chords, 
+                       input_number_prime_chords,
+                       input_number_gen_chords,
                        input_use_original_durations,
                        input_match_original_pitches_counts,
                        input_number_prime_tokens,
@@ -51,6 +52,7 @@ def Generate_Rock_Song(input_midi,
     print('Input MIDI file name:', fn)
     print('Generation type:', input_gen_type)
     print('Number of prime chords:', input_number_prime_chords)
+    print('Number of chords to generate:', input_number_gen_chords)
     print('Use original durations:', input_use_original_durations)
     print('Match original pitches counts:', input_match_original_pitches_counts)
     print('Number of prime tokens:', input_number_prime_tokens)
@@ -267,7 +269,7 @@ def Generate_Rock_Song(input_midi,
         for i in range(input_number_prime_chords):
             song.extend(prime_toks[i])
         
-        for i in tqdm.tqdm(range(input_number_prime_chords, len(score_toks))):
+        for i in tqdm.tqdm(range(input_number_prime_chords, input_number_prime_chords+input_number_gen_chords)):
             
             song.extend(score_toks[i])
         
@@ -412,13 +414,14 @@ if __name__ == "__main__":
         gr.Markdown("## Controlled generation options")
 
         input_number_prime_chords = gr.Slider(0, 512, value=128, step=8, label="Number of prime chords")
+        input_number_gen_chords = gr.Slider(16, 1024, value=512, step=8, label="Number of chords to generate")
         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)
 
         gr.Markdown("## Freestyle continuation options")
 
         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=1024, step=16, label="Number of tokens to generate")
+        input_number_gen_tokens = gr.Slider(0, 3072, value=1024, step=16, label="Number of tokens to generate")
 
         gr.Markdown("## Model options")
 
@@ -438,7 +441,8 @@ if __name__ == "__main__":
 
         run_event = run_btn.click(Generate_Rock_Song, [input_midi,
                                                        input_gen_type,
-                                                       input_number_prime_chords, 
+                                                       input_number_prime_chords,
+                                                       input_number_gen_chords,
                                                        input_use_original_durations,
                                                        input_match_original_pitches_counts,
                                                        input_number_prime_tokens,
@@ -450,13 +454,14 @@ if __name__ == "__main__":
                                   [output_midi_title, output_midi_summary, output_midi, output_audio, output_plot])
 
         gr.Examples(
-            [["Sharing The Night Together.kar", "Controlled", 256, True, True, False, 512, 1024, 2048, 0.9, 10],
-             ["Hotel California.mid", "Controlled", 256, True, True, False, 512, 1024, 2048, 0.9, 10],
-             ["Nothing Else Matters.kar", "Controlled", 256, True, True, False, 512, 1024, 2048, 0.9, 10],
+            [["Sharing The Night Together.kar", "Controlled", 256, 512, True, True, False, 512, 1024, 2048, 0.9, 10],
+             ["Hotel California.mid", "Controlled", 256, 512, True, True, False, 512, 1024, 2048, 0.9, 10],
+             ["Nothing Else Matters.kar", "Controlled", 256, 512, True, True, False, 512, 1024, 2048, 0.9, 10],
             ],
             [input_midi,
              input_gen_type,
-             input_number_prime_chords, 
+             input_number_prime_chords,
+             input_number_gen_chords,
              input_use_original_durations,
              input_match_original_pitches_counts,
              input_number_prime_tokens,