Update app.py

app.py

@@ -510,29 +510,25 @@ if __name__ == "__main__":
         
         input_midi = gr.File(label="Input MIDI", file_types=[".midi", ".mid", ".kar"])
         
-        gr.Markdown("## Select generation type")
+        gr.Markdown("## Select conversion type")
 
-        input_gen_type = gr.Radio(["Controlled", "Freestyle"], value='Controlled', label="Generation type")
-
-        gr.Markdown("## Controlled generation options")
-
-        input_number_prime_chords = gr.Slider(0, 512, value=0, step=8, label="Number of prime chords")
-        input_number_gen_chords = gr.Slider(16, 512, value=256, step=8, label="Number of chords to generate")
-        input_use_original_durations = gr.Checkbox(label="Use original durations", value=True)
-        input_match_original_pitches_counts = gr.Checkbox(label="Match original pitches counts", value=True)
+        input_conv_type = gr.Radio(["Durations and Velocities", "Durations", "Velocities"], 
+                                   value="Durations and Velocities", 
+                                   label="Conversion type"
+                                  )
 
-        gr.Markdown("## Freestyle continuation options")
+        gr.Markdown("## Conversion options")
 
-        input_number_prime_tokens = gr.Slider(0, 1024, value=512, step=16, label="Number of prime tokens")
-        input_number_gen_tokens = gr.Slider(0, 3072, value=1024, step=16, label="Number of tokens to generate")
+        input_number_prime_notes = gr.Slider(0, 512, value=0, step=8, label="Number of prime notes")
+        input_number_conv_notes = gr.Slider(0, 3072, value=1024, step=16, label="Number of notes to convert")
 
         gr.Markdown("## Model options")
 
-        input_num_memory_tokens = gr.Slider(1024, 4096, value=2048, step=16, label="Number of memory tokens")
-        input_model_temperature = gr.Slider(0.1, 1.0, value=0.9, step=0.05, label="Model temperature")
-        input_model_top_p = gr.Slider(0.1, 1, value=0.96, step=0.01, label="Model sampling top p value")
+        input_model_dur_top_k = gr.Slider(1, 100, value=1, step=1, label="Model sampling top k value for durations")
+        input_model_dur_temperature = gr.Slider(0.5, 1.5, value=1.1, step=0.05, label="Model temperature for durations")
+        input_model_vel_temperature = gr.Slider(0.5, 1.5, value=1.5, step=0.05, label="Model temperature for velocities")
         
-        run_btn = gr.Button("generate", variant="primary")
+        run_btn = gr.Button("convert", variant="primary")
 
         gr.Markdown("## Generation results")
 
@@ -543,40 +539,32 @@ if __name__ == "__main__":
         output_midi = gr.File(label="Output MIDI file", file_types=[".mid"])
 
         run_event = run_btn.click(Convert_Score_to_Performance, [input_midi,
-                                                               input_gen_type,
-                                                               input_number_prime_chords,
-                                                               input_number_gen_chords,
-                                                               input_use_original_durations,
-                                                               input_match_original_pitches_counts,
-                                                               input_number_prime_tokens,
-                                                               input_number_gen_tokens,
-                                                               input_num_memory_tokens,
-                                                               input_model_temperature,
-                                                               input_model_top_p,
-                                                              ],
+                                                                 input_conv_type,
+                                                                 input_number_prime_notes,
+                                                                 input_number_conv_notes,
+                                                                 input_model_dur_top_k,
+                                                                 input_model_dur_temperature,
+                                                                 input_model_vel_temperature
+                                                                ],
                                   [output_midi_title, output_midi_summary, output_midi, output_audio, output_plot])
 
         gr.Examples(
-            [["asap_midi_score_21.mid", "Controlled", 0, 512, True, True, 512, 1024, 2048, 0.9, 0.96],
-             ["asap_midi_score_45.mid", "Controlled", 128, 256, False, False, 512, 1024, 2048, 0.9, 0.96],
-             ["asap_midi_score_69.mid", "Controlled", 128, 256, True, True, 512, 1024, 2048, 0.9, 0.96],
-             ["asap_midi_score_118.mid", "Controlled", 128, 256, True, True, 512, 1024, 2048, 0.9, 0.96],
-             ["asap_midi_score_167.mid", "Controlled", 128, 256, True, True, 512, 1024, 2048, 0.9, 0.96],
+            [["asap_midi_score_21.mid", "Durations and Velocities", 8, 600, 1, 1.1, 1.5],
+             ["asap_midi_score_45.mid", "Durations and Velocities", 8, 600, 1, 1.1, 1.5],
+             ["asap_midi_score_69.mid", "Durations and Velocities", 8, 600, 1, 1.1, 1.5],
+             ["asap_midi_score_118.mid", "Durations and Velocities", 8, 600, 1, 1.1, 1.5],
+             ["asap_midi_score_167.mid", "Durations and Velocities", 8, 600, 1, 1.1, 1.5],
             ],
             [input_midi,
-             input_gen_type,
-             input_number_prime_chords,
-             input_number_gen_chords,
-             input_use_original_durations,
-             input_match_original_pitches_counts,
-             input_number_prime_tokens,
-             input_number_gen_tokens,
-             input_num_memory_tokens,
-             input_model_temperature,
-             input_model_top_p,
+             input_conv_type,
+             input_number_prime_notes,
+             input_number_conv_notes,
+             input_model_dur_top_k,
+             input_model_dur_temperature,
+             input_model_vel_temperature
             ],
             [output_midi_title, output_midi_summary, output_midi, output_audio, output_plot],
-            Generate_Rock_Song,
+            Convert_Score_to_Performance,
             cache_examples=True,
         )