Update app.py

app.py

@@ -205,11 +205,11 @@ def Generate_Music_Bridge(input_midi,
 
         print('Loading MIDI...')
     
-        score = load_midi(input_midi.name)
+        melody_chords = load_midi(input_midi.name)
     
-        if score is not None:
+        if melody_chords is not None:
             
-            print('Sample score tokens', score[:10])
+            print('Sample score tokens', melody_chords[:10])
         
             #==================================================================
             
@@ -435,21 +435,18 @@ with gr.Blocks() as demo:
     
     gr.Markdown("## Generation options")
 
+    patches_to_inpaint = gr.Dropdown(label="Select instruments to inpaint", choices=list(patch2number.keys()),
+                                     multiselect=True, type="value",
+                                     info="Instruments MUST be present in the composition. For best results select a single instrument."
+                                    )
 
-    patches_to_inpaint = [0]
-    inpaint_every_nth_note = 1
-    forbid_original_pitches = True
-    max_inpainted_pitch_dev = 12
-    max_inpaint_tries_per_note = 20
-    
-    num_prime_tokens = 128
-    num_mem_tokens = 4096
-
+    inpaint_every_nth_note = gr.Slider(1, 10, value=1, step=1, label="Inpaint every nth note")
+    forbid_original_pitches = gr.Checkbox(value=True, label="Forbid repeating original pitches")
+    max_inpainted_pitch_dev = gr.Slider(12, 24, value=12, step=12, label="Maximum inpainted pitch deviation")
+    max_inpaint_tries_per_note = gr.Slider(5, 100, value=20, step=1, label="Maximum inpainting attempts per note")
 
-    inpaint_instruments = gr.Dropdown(label="Select instruments to inpaint", choices=list(patch2number.keys()),
-                                      multiselect=True, type="value",
-                                      info="Instruments MUST be present in the composition. For best results select a single instrument."
-                                     )
+    num_prime_tokens = gr.Slider(0, 512, value=128, step=1, label="Number of prime tokens")
+    num_mem_tokens = gr.Slider(32, 8192, value=4096, step=8, label="Number of prime tokens")
 
     model_temperature = gr.Slider(0.1, 1, value=0.9, step=0.01, label="Model temperature")
     model_sampling_top_k = gr.Slider(1, 100, value=15, step=1, label="Model sampling top k value")