skytnt commited on
Commit
bd059e5
·
verified ·
1 Parent(s): 639d648

Update app_onnx.py

Browse files
Files changed (1) hide show
  1. app_onnx.py +3 -1
app_onnx.py CHANGED
@@ -450,6 +450,8 @@ key_signatures = ['C♭', 'A♭m', 'G♭', 'E♭m', 'D♭', 'B♭m', 'A♭', 'Fm
450
 
451
  if __name__ == "__main__":
452
  parser = argparse.ArgumentParser()
 
 
453
  parser.add_argument("--device", type=str, default="cuda", help="device to run model")
454
  parser.add_argument("--batch", type=int, default=8, help="batch size")
455
  parser.add_argument("--max-gen", type=int, default=1024, help="max")
@@ -622,5 +624,5 @@ if __name__ == "__main__":
622
  # queue=False)
623
  undo_btn.click(undo_continuation, [input_model, output_midi_seq, output_continuation_state],
624
  [output_midi_seq, output_continuation_state, js_msg], queue=False)
625
- app.queue().launch(inbrowser=True, ssr_mode=False)
626
  thread_pool.shutdown()
 
450
 
451
  if __name__ == "__main__":
452
  parser = argparse.ArgumentParser()
453
+ parser.add_argument("--share", action="store_true", default=False, help="share gradio app")
454
+ parser.add_argument("--port", type=int, default=7860, help="gradio server port")
455
  parser.add_argument("--device", type=str, default="cuda", help="device to run model")
456
  parser.add_argument("--batch", type=int, default=8, help="batch size")
457
  parser.add_argument("--max-gen", type=int, default=1024, help="max")
 
624
  # queue=False)
625
  undo_btn.click(undo_continuation, [input_model, output_midi_seq, output_continuation_state],
626
  [output_midi_seq, output_continuation_state, js_msg], queue=False)
627
+ app.queue().launch(server_port=opt.port, share=opt.share, inbrowser=True, ssr_mode=False)
628
  thread_pool.shutdown()