Update app.py

app.py

@@ -456,6 +456,7 @@ def reset(final_composition=[], generated_batches=[], block_lines=[]):
     return [], [], []
 
 patch2number = {v: k for k, v in TMIDIX.Number2patch.items()}
+patch2number['Drums'] = 128
 
 # -----------------------------
 # GRADIO INTERFACE SETUP
@@ -518,7 +519,9 @@ with gr.Blocks() as demo:
 
     gr.Markdown("## Generate")
     prime_instruments = gr.Dropdown(label="Prime instruments (select up to 5)", choices=list(patch2number.keys()),
-                                    multiselect=True, max_choices=5, type="value")
+                                    multiselect=True, max_choices=5, type="value",
+                                    info="Instruments are asigned in order of selection from top to bottom. Custom MIDI overrides prime instruments."
+                                   )
     num_prime_tokens = gr.Slider(16, 7168, value=7168, 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")