Update app.py

app.py

@@ -322,6 +322,13 @@ def generate_music_and_state(input_midi, num_prime_tokens, num_gen_tokens, num_m
         midi_fname, midi_score = save_midi(final_composition)
         # Use the last note's time as a marker.
         block_lines.append(midi_score[-1][1] / 1000 if final_composition else 0)
+
+    if add_outro:
+        final_composition.append(18817) # Outro token
+
+    if add_drums:
+        pass
+        
     
     batched_gen_tokens = generate_music(final_composition, num_gen_tokens, num_mem_tokens,
                                         NUM_OUT_BATCHES, model_temperature)