Update app.py

app.py

@@ -326,7 +326,9 @@ def generate_callback(input_midi,
 
     if not final_composition and input_midi is not None:
         final_composition.extend(load_midi(input_midi)[:num_prime_tokens])
-
+        midi_score = save_midi(final_composition)
+        block_lines.append(midi_score[-1][1] / 1000)
+        
     batched_gen_tokens = generate_music(final_composition, 
                                         num_gen_tokens, 
                                         NUM_OUT_BATCHES,
@@ -427,10 +429,6 @@ def generate_callback_wrapper(input_midi,
 #==================================================================================
 
 def add_batch(batch_number):
-
-    if not block_lines:
-        midi_score = save_midi(final_composition)
-        block_lines.append(midi_score[-1][1] / 1000)
     
     final_composition.extend(generated_batches[batch_number])