Update app.py

app.py

@@ -206,7 +206,7 @@ if __name__ == "__main__":
     parser = argparse.ArgumentParser()
     parser.add_argument("--share", action="store_true", default=False, help="share gradio app")
     parser.add_argument("--port", type=int, default=7860, help="gradio server port")
-    parser.add_argument("--max-gen", type=int, default=512, help="max")
+    parser.add_argument("--max-gen", type=int, default=1024, help="max")
     opt = parser.parse_args()
     soundfont_path = hf_hub_download(repo_id="skytnt/midi-model", filename="soundfont.sf2")
     model_base_path = hf_hub_download(repo_id="skytnt/midi-model", filename="onnx/model_base.onnx")
@@ -257,17 +257,17 @@ if __name__ == "__main__":
         tab1.select(lambda: 0, None, tab_select, queue=False)
         tab2.select(lambda: 1, None, tab_select, queue=False)
         input_gen_events = gr.Slider(label="generate n midi events", minimum=1, maximum=opt.max_gen,
-                                     step=1, value=opt.max_gen)
+                                     step=1, value=opt.max_gen // 2)
         input_temp = gr.Slider(label="temperature", minimum=0.1, maximum=1.2, step=0.01, value=1)
         input_top_p = gr.Slider(label="top p", minimum=0.1, maximum=1, step=0.01, value=0.98)
         input_top_k = gr.Slider(label="top k", minimum=1, maximum=50, step=1, value=20)
         input_allow_cc = gr.Checkbox(label="allow midi cc event", value=True)
         run_btn = gr.Button("generate", variant="primary")
-        stop_btn = gr.Button("stop")
+        stop_btn = gr.Button("stop and output")
         output_midi_seq = gr.Variable()
         output_midi_img = gr.Image(label="output image")
         output_midi = gr.File(label="output midi", file_types=[".mid"])
-        output_audio = gr.Audio(label="output audio", format="wav")
+        output_audio = gr.Audio(label="output audio", format="mp3")
         run_event = run_btn.click(run, [tab_select, input_instruments, input_drum_kit, input_midi, input_midi_events,
                                         input_gen_events, input_temp, input_top_p, input_top_k,
                                         input_allow_cc],