Update app.py

app.py

@@ -296,8 +296,8 @@ def generate_music(prime, num_gen_tokens, num_gen_batches, model_temperature, mo
     
     """Generate music tokens given prime tokens and parameters."""
 
-    if len(prime) >= 7168:
-        prime = [18816] + prime[-7168:]
+    if len(prime) >= 6656:
+        prime = [18816] + prime[-6656:]
     
     inputs = prime
     
@@ -372,7 +372,7 @@ def generate_music_and_state(input_midi,
                                       remove_overlapping_durations=remove_overlapping_durations
                                      )
 
-        if num_prime_tokens < 7168:
+        if num_prime_tokens < 6656:
             final_composition = final_composition[:num_prime_tokens]
         
         midi_fname, midi_score = save_midi(final_composition)
@@ -574,7 +574,7 @@ with gr.Blocks() as demo:
     prime_instruments.input(reset, [final_composition, generated_batches, block_lines],
                           [final_composition, generated_batches, block_lines])
     
-    num_prime_tokens = gr.Slider(16, 7168, value=7168, step=1, label="Number of prime tokens")
+    num_prime_tokens = gr.Slider(16, 6656, value=6656, step=1, label="Number of prime tokens")
     num_gen_tokens = gr.Slider(16, 1024, value=512, step=1, label="Number of tokens to generate")
     model_temperature = gr.Slider(0.1, 1, value=0.9, step=0.01, label="Model temperature")
     model_top_p = gr.Slider(0.1, 0.99, value=0.96, step=0.01, label="Model sampling top p value")