diff --git a/app.py b/app.py
index c2237b5..f0f94d0 100644
--- a/app.py
+++ b/app.py
@@ -24,6 +24,8 @@ with Path("configs/paths.yml").open("r", encoding="utf-8") as f:
 
 parser = argparse.ArgumentParser()
 parser.add_argument("--device", type=str, default="cuda")
+parser.add_argument("--host", type=str, default="127.0.0.1")
+parser.add_argument("--port", type=int, default=7860)
 parser.add_argument("--no_autolaunch", action="store_true")
 parser.add_argument("--share", action="store_true")
 
@@ -49,4 +51,4 @@ with gr.Blocks(theme=GRADIO_THEME) as app:
             create_merge_app(model_holder=model_holder)
 
 
-app.launch(inbrowser=not args.no_autolaunch, share=args.share)
+app.launch(server_name=args.host, server_port=args.port, inbrowser=not args.no_autolaunch, share=args.share)
diff --git a/webui/train.py b/webui/train.py
index fa29719..d837471 100644
--- a/webui/train.py
+++ b/webui/train.py
@@ -792,5 +792,4 @@ def create_train_app():
             outputs=[use_jp_extra_train],
         )
 
-    # app.launch(inbrowser=not args.no_autolaunch, server_name=args.server_name)
     return app