Change default based on statistics

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 = ["A2", "D3", "G3", "C4", "E4"]
-notes_to_midi = {"A2": 45, "D3": 50, "G3": 55, "C4": 60, "E4": 64}
+notes = ["A2", "C3", "D3", "E4", "G4", "E5"]
+notes_to_midi = {"A2": 45, "C3": 48, "D3": 50, "E4": 64, "G4":67, "E5": 76}
 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"}
 
@@ -81,9 +81,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="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")
+    time_signature = gr.Dropdown(time_signatures, value="2/4", label="Time signature") 
+    note = gr.Dropdown(notes, value="E4", label="First note")
+    bpm = gr.Slider(minimum=60, maximum=130, step=10, value=90, label="Tempo")
     density = gr.Slider(minimum=0, maximum=4, step=1, value=2, label="Density")
   with gr.Row():
     btn = gr.Button("Compose")