Update app.py

app.py

@@ -396,25 +396,28 @@ if __name__ == "__main__":
             "![Visitors](https://api.visitorbadge.io/api/visitors?path=asigalov61.Guided-Rock-Music-Transformer&style=flat)\n\n")
 
         gr.Markdown("## Upload your MIDI or select a sample example MIDI below")
+        gr.Markdown("### For best results use MIDIs with 1:2 notes to drums ratio")
         
         input_midi = gr.File(label="Input MIDI", file_types=[".midi", ".mid", ".kar"])
         
-        gr.Markdown("## Select generation options")
+        gr.Markdown("## Select generation type")
 
-        gr.Markdown("### Controlled generation options")
+        input_gen_type = gr.Radio(["Controlled", "Freestyle"], label="Generation type"),
+
+        gr.Markdown("## Controlled generation options")
 
         input_number_prime_chords = gr.Slider(0, 512, value=128, step=8, label="Number of prime chords")
         input_use_original_durations = gr.Checkbox(label="Use original durations", value=False)
         input_match_original_pitches_counts = gr.Checkbox(label="Match original pitches counts", value=False)
 
-        gr.Markdown("### Freestyle continuation options")
+        gr.Markdown("## Freestyle continuation options")
 
-        input_freestyle_continuation = gr.Checkbox(label="Freestyle continuation", value=False)
         input_number_prime_tokens = gr.Slider(0, 1024, value=512, step=16, label="Number of prime tokens")
         input_number_gen_tokens = gr.Slider(0, 2048, value=1024, step=16, label="Number of tokens to generate")
 
-        gr.Markdown("### Model options")
+        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_k = gr.Slider(1, 50, value=10, step=1, label="Model sampling top k value")