Update app.py

app.py

@@ -22,7 +22,11 @@ in_space = os.getenv("SYSTEM") == "spaces"
 @torch.no_grad()
 def GenerateMIDI(num_tok, idrums, iinstr, progress=gr.Progress()):
 
+    print('=' * 70)
     print('Req num tok', num_tok)
+    print('Req instr', iinstr)
+    print('Drums', idrums)
+    print('=' * 70)
 
     if idrums:
         drums = 3074
@@ -56,9 +60,6 @@ def GenerateMIDI(num_tok, idrums, iinstr, progress=gr.Progress()):
 
     melody_chords_f = outy
     
-    print('=' * 70)
-    print('Done!')
-    print('=' * 70)
     print('Sample INTs', melody_chords_f[:12])
     print('=' * 70)
     
@@ -210,7 +211,7 @@ if __name__ == "__main__":
         
         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")
+        input_num_tokens = gr.Slider(16, 256, value=128, label="Number of Tokens", info="Number of tokens to generate")
         run_btn = gr.Button("generate", variant="primary")
 
         output_midi_seq = gr.Variable()