Add more options to the inputs

app.py

@@ -9,8 +9,8 @@ SAMPLE_RATE=44100
 
 # Feel free to change this, I am using only three notes here because the model 
 # works better this way.
-notes = ["D3", "G3", "E4"]
-notes_to_midi = {"D3": 50, "G3": 55, "E4": 64}
+notes = ["A2", "D3", "G3", "C4", "E4"]
+notes_to_midi = {"A2": 45, "D3": 50, "G3": 55, "C4": 60, "E4": 64}
 time_signatures = ["4/4", "3/4", "2/4", "6/8"]
 time_signature_to_tokens = {"4/4": "4_4", "3/4": "3_4", "2/4": "2_4", "6/8": "6_8"}
 
@@ -80,9 +80,9 @@ with demo:
   # UI for the inputs of the model
   gr.Markdown("Select the generation parameters.")
   with gr.Row():
-    time_signature = gr.Dropdown(time_signatures, value="4/4", label="Time signature") 
-    note = gr.Dropdown(notes, value="G3", label="First note")
-    bpm = gr.Slider(minimum=60, maximum=140, step=10, value=90, label="Tempo")
+    time_signature = gr.Dropdown(time_signatures, value="6/8", label="Time signature") 
+    note = gr.Dropdown(notes, value="D3", label="First note")
+    bpm = gr.Slider(minimum=60, maximum=140, step=10, value=120, label="Tempo")
     density = gr.Slider(minimum=0, maximum=4, step=1, value=2, label="Density")
   with gr.Row():
     btn = gr.Button("Compose")