Update app.py

app.py

@@ -20,9 +20,9 @@ in_space = os.getenv("SYSTEM") == "spaces"
 #=================================================================================================
 
 @torch.no_grad()
-def GenerateMIDI(idrums, iinstr, progress=gr.Progress()):
+def GenerateMIDI(num_tok, idrums, iinstr, progress=gr.Progress()):
 
-    number_of_notes_to_generate = 140
+    print('Req num tok', num_tok)
 
     if idrums:
         drums = 3074
@@ -40,12 +40,12 @@ def GenerateMIDI(idrums, iinstr, progress=gr.Progress()):
     
     outy = start_tokens
 
-    for i in progress.tqdm(range(number_of_notes_to_generate)):
+    for i in progress.tqdm(range(num_tok)):
     
         inp = torch.LongTensor([outy]).cpu()
         
         out = model.module.generate(inp,
-                              3,
+                              1,
                               temperature=0.9,
                               return_prime=False,
                               verbose=True)
@@ -205,14 +205,16 @@ if __name__ == "__main__":
                     "(https://colab.research.google.com/github/asigalov61/Allegro-Music-Transformer/blob/main/Allegro_Music_Transformer_Composer.ipynb)"
                     " for faster execution and endless generation"
                         )
+        
         input_drums = gr.Checkbox(label="Drums Controls", value = False, info="Drums present or not")
         input_instrument = gr.Radio(["Piano", "Guitar", "Bass", "Violin", "Cello", "Harp", "Trumpet", "Sax", "Flute", "Choir", "Organ"], value="Piano", label="Lead Instrument Controls", info="Desired lead instrument")       
+        input_num_tokens = gr.Slider(16, 512, value=256, label="Number of Tokens", info="Number of tokens to generate")
         run_btn = gr.Button("generate", variant="primary")
 
         output_midi_seq = gr.Variable()
         output_audio = gr.Audio(label="output audio", format="mp3", elem_id="midi_audio")
         output_plot = gr.Plot(label="output plot")
         output_midi = gr.File(label="output midi", file_types=[".mid"])
-        run_event = run_btn.click(GenerateMIDI, [input_drums, input_instrument], [output_midi_seq, output_plot, output_midi, output_audio])
+        run_event = run_btn.click(GenerateMIDI, [input_num_tokens, input_drums, input_instrument], [output_midi_seq, output_plot, output_midi, output_audio])
         
         app.queue(concurrency_count=1).launch(server_port=opt.port, share=opt.share, inbrowser=True)