Spaces:
Running
on
Zero
Running
on
Zero
Update app_onnx.py
Browse files- 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()
|