Update app.py

app.py

@@ -274,6 +274,10 @@ def save_midi(tokens):
 @spaces.GPU
 def generate_music(prime, num_gen_tokens, num_gen_batches, model_temperature, model_top_p):
     """Generate music tokens given prime tokens and parameters."""
+
+    if len(prime) > 7168:
+        prime = [18816] + prime[-7168:]
+    
     inputs = prime if prime else [18816]
     print("Generating...")
     inp = torch.LongTensor([inputs] * num_gen_batches).cuda()
@@ -413,12 +417,16 @@ def remove_batch(batch_number, num_tokens, final_composition, generated_batches,
 
 def clear():
     """Clear outputs and reset state."""
+    print_sep()
     print('Clear batch...')
+    print_sep()
     return None, None, None, [], []
 
 def reset(final_composition=[], generated_batches=[], block_lines=[]):
     """Reset composition state."""
-    print('Reset...')
+    print_sep()
+    print('Reset MIDI...')
+    print_sep()
     return [], [], []
 
 # -----------------------------