diff --git a/demo_gradio.py b/demo_gradio.py index 632cb87..687cf56 100644 --- a/demo_gradio.py +++ b/demo_gradio.py @@ -31,9 +31,13 @@ from diffusers_helper.bucket_tools import find_nearest_bucket parser = argparse.ArgumentParser() parser.add_argument('--share', action='store_true') parser.add_argument("--server", type=str, default='0.0.0.0') -parser.add_argument("--port", type=int, default=7860) +parser.add_argument("--port", type=int, required=False) +parser.add_argument("--inbrowser", action='store_true') args = parser.parse_args() +# for win desktop probably use --server 127.0.0.1 --inbrowser +# For linux server probably use --server 127.0.0.1 or do not use any cmd flags + print(args) free_mem_gb = get_cuda_free_memory_gb(gpu) @@ -396,4 +400,5 @@ block.launch( server_name=args.server, server_port=args.port, share=args.share, + inbrowser=args.inbrowser, )