Update app.py

app.py

@@ -199,7 +199,8 @@ def save_midi(tokens, batch_number=None, model_selector=''):
 
 @spaces.GPU
 def Generate_Accompaniment(input_midi, 
-                           num_gen_tokens,
+                           generation_type,
+                           melody_patch,
                            model_temperature
                           ):
 
@@ -372,7 +373,8 @@ with gr.Blocks() as demo:
     
     gr.Markdown("## Generation options")
     
-    generation_type = gr.Radio(["Guided", "Freestyle"] value="Guided", step=1, label="Generation type")
+    generation_type = gr.Radio(["Guided", "Freestyle"], value="Guided", step=1, label="Generation type")
+    melody_patch = gr.Slider(-1, 127, value=-1, step=1, label="Source melody MIDI patch")
     model_temperature = gr.Slider(0.1, 1, value=0.9, step=0.01, label="Model temperature")
     
     generate_btn = gr.Button("Generate", variant="primary")
@@ -386,6 +388,7 @@ with gr.Blocks() as demo:
     generate_btn.click(Generate_Accompaniment, 
                        [input_midi, 
                         generation_type,
+                        melody_patch,
                         model_temperature
                        ], 
                        [