Update app.py

app.py

@@ -118,7 +118,7 @@ def GenerateMIDI(num_tok, idrums, iinstr):
     pitch = 0
     channel = 0
 
-    for i in range(max(1, min(512, num_tok))):
+    for i in range(max(1, min(1024, num_tok))):
         
         inp = torch.LongTensor([outy]).cuda()
 
@@ -154,7 +154,7 @@ def GenerateMIDI(num_tok, idrums, iinstr):
 
     output_plot = TMIDIX.plot_ms_SONG(output[2], plot_title='Allegro-Music-Transformer-Composition', return_plt=True)
 
-    audio = midi_to_colab_audio(new_fn, 
+    audio = midi_to_colab_audio('Allegro-Music-Transformer-Composition.mid', 
                         soundfont_path="SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2",
                         sample_rate=16000,
                         volume_scale=10,
@@ -197,7 +197,7 @@ if __name__ == "__main__":
         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")
+        input_num_tokens = gr.Slider(16, 1024, value=512, label="Number of Tokens", info="Number of tokens to generate")
         run_btn = gr.Button("generate", variant="primary")
 
         output_plot = gr.Plot(label='output plot')