Update app.py

app.py

@@ -445,13 +445,6 @@ with gr.Blocks() as demo:
             """)
     
     #==================================================================================
-
-    final_composition = gr.State([])
-    generated_batches = gr.State([])
-    block_lines = gr.State([])
-    model_state = gr.State([])
-    
-    #==================================================================================
     
     gr.Markdown("## Upload seed MIDI or click 'Generate' button for random output")
     
@@ -460,51 +453,30 @@ with gr.Blocks() as demo:
                             [final_composition, generated_batches, block_lines])
     
     gr.Markdown("## Generate")
-
-    model_selector = gr.Dropdown(["without velocity - 7 epochs",
-                                  "without velocity - 3 epochs", 
-                                  "with velocity - 3 epochs"
-                                ], 
-                                label="Select model", 
-                               )
     
-    num_prime_tokens = gr.Slider(15, 1024, value=1024, step=1, label="Number of prime tokens")
     num_gen_tokens = gr.Slider(15, 1024, value=1024, step=1, label="Number of tokens to generate")
-    num_mem_tokens = gr.Slider(15, 2048, value=2048, step=1, label="Number of memory tokens")
     model_temperature = gr.Slider(0.1, 1, value=0.9, step=0.01, label="Model temperature")
-    # model_sampling_top_p = gr.Slider(0.1, 1, value=0.96, step=0.01, label="Model sampling top p value")
     
     generate_btn = gr.Button("Generate", variant="primary")
 
-    gr.Markdown("## Select batch")
+    gr.Markdown("## Results")
     
-    outputs = [final_composition, generated_batches, block_lines]
-    
-    for i in range(NUM_OUT_BATCHES):
-        with gr.Tab(f"Batch # {i}") as tab:
-            
-            audio_output = gr.Audio(label=f"Batch # {i} MIDI Audio", format="mp3", elem_id="midi_audio")
-            plot_output = gr.Plot(label=f"Batch # {i} MIDI Plot")
-            midi_output = gr.File(label=f"Batch # {i} MIDI File")
+    output_audio = gr.Audio(label="Output MIDI audio", format="wav", elem_id="midi_audio")
+    output_plot = gr.Plot(label="Output MIDI score plot")
+    output_midi = gr.File(label="Output MIDI file", file_types=[".mid"])
             
-            outputs.extend([audio_output, plot_output, midi_output])
-
     outputs.extend([model_state])
 
-    generate_btn.click(generate_callback_wrapper, 
+    generate_btn.click(generate_accompaniment, 
                        [input_midi, 
-                        num_prime_tokens, 
                         num_gen_tokens,
-                        num_mem_tokens,
                         model_temperature,
-                        # model_sampling_top_p,
-                        final_composition,
-                        generated_batches,
-                        block_lines,
-                        model_selector,
-                        model_state
                        ], 
-                       outputs
+                       [
+                        output_audio,
+                        output_plot,
+                        output_midi,                           
+                       ]
                       )
     
     #==================================================================================