support sys.argv --listen --share --port
This commit is contained in:
lllyasviel
2023-08-14 06:56:23 -07:00
committed by GitHub
parent 50708f3d22
commit 7900480360
5 changed files with 25 additions and 4 deletions
+8 -1
View File
@@ -2,6 +2,7 @@ import gradio as gr
import sys
import time
import shared
import argparse
import modules.path
import fooocus_version
import modules.html
@@ -94,4 +95,10 @@ with shared.gradio_root:
ctrls += [base_model, refiner_model] + lora_ctrls
run_button.click(fn=generate_clicked, inputs=ctrls, outputs=[run_button, progress_html, progress_window, gallery])
shared.gradio_root.launch(inbrowser=True, server_name='0.0.0.0' if '--listen' in sys.argv else None)
parser = argparse.ArgumentParser()
parser.add_argument("--listen", action='store_true', help="Listen remote port.")
parser.add_argument("--port", type=int, default=None, help="Set the listen port.")
parser.add_argument("--share", action='store_true', help="Set whether to share on Gradio.")
args = parser.parse_args()
shared.gradio_root.launch(inbrowser=True, server_name='0.0.0.0' if args.listen else None, server_port=args.port, share=args.share)