diff --git a/src/backend/main.py b/src/backend/main.py
index e51b109a..e06d2e81 100644
--- a/src/backend/main.py
+++ b/src/backend/main.py
@@ -150,6 +150,7 @@ async def serve_index():
     if model_name is not None and init_nodes_num is not None:
         scheduler_manage.run(model_name, init_nodes_num, is_local_network)
 
+    host = args.host
     port = args.port
 
-    uvicorn.run(app, host="localhost", port=port, log_level="info", loop="uvloop")
+    uvicorn.run(app, host=host, port=port, log_level="info", loop="uvloop")
diff --git a/src/backend/server/server_args.py b/src/backend/server/server_args.py
index 27207c7c..a5179821 100644
--- a/src/backend/server/server_args.py
+++ b/src/backend/server/server_args.py
@@ -21,6 +21,7 @@ def parse_args() -> argparse.Namespace:
     parser.add_argument("--dht-prefix", type=str, default="gradient", help="Prefix for DHT keys")
 
     # Scheduler configuration
+    parser.add_argument("--host", type=str, default="localhost", help="Host to listen on")
     parser.add_argument("--port", type=int, default=3001, help="Port to listen on")
     parser.add_argument(
         "--log-level",