Update app.py

app.py

@@ -353,6 +353,8 @@ key_signatures = ['C♭', 'A♭m', 'G♭', 'E♭m', 'D♭', 'B♭m', 'A♭', 'Fm
 
 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("--device", type=str, default="cuda", help="device to run model")
     parser.add_argument("--batch", type=int, default=8, help="batch size")
     parser.add_argument("--max-gen", type=int, default=1024, help="max")
@@ -527,5 +529,5 @@ if __name__ == "__main__":
         #                queue=False)
         undo_btn.click(undo_continuation, [input_model, output_midi_seq, output_continuation_state],
                        [output_midi_seq, output_continuation_state, js_msg], queue=False)
-    app.queue().launch(ssr_mode=False)
+    app.queue().launch(server_port=opt.port, share=opt.share, ssr_mode=False)
     thread_pool.shutdown()