Update app.py

app.py

@@ -180,6 +180,7 @@ if __name__ == "__main__":
     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("--device", type=str, default="cpu", help="device to run model")
+    parser.add_argument("--max-gen", type=int, default=512, help="max")
     soundfont_path = hf_hub_download(repo_id="skytnt/midi-model", filename="soundfont.sf2")
     model_path = hf_hub_download(repo_id="skytnt/midi-model", filename="model.ckpt")
     opt = parser.parse_args()
@@ -192,6 +193,11 @@ if __name__ == "__main__":
 
     app = gr.Blocks()
     with app:
+        gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Midi Composer</h1>")
+        gr.Markdown("![Visitors](https://api.visitorbadge.io/api/visitors?path=skytnt.midi-composer&style=flat)\n\n"
+                    "Midi event transformer for music generation\n\n"
+                    "Demo for [SkyTNT/midi-model](https://github.com/SkyTNT/midi-model)")
+
         tab_select = gr.Variable(value=0)
         with gr.Tabs():
             with gr.TabItem("instrument prompt") as tab1:
@@ -207,7 +213,8 @@ 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=512, step=1, value=512)
+        input_gen_events = gr.Slider(label="generate n midi events", minimum=1, maximum=opt.max_gen,
+                                     step=1, value=opt.max_gen)
         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.97)
         input_top_k = gr.Slider(label="top k", minimum=1, maximum=50, step=1, value=20)