update app.py

app.py

@@ -333,6 +333,8 @@ def http_bot(
         controller_url + "/get_worker_address", json={"model": model_name}
     )
     worker_addr = ret.json()["address"]
+    if worker_addr.startswith("http://0.0.0.0"):
+        worker_addr = worker_addr.replace("0.0.0.0", args.worker_ip)
     logger.info(f"model_name: {model_name}, worker_addr: {worker_addr}")
 
     # No available worker
@@ -825,6 +827,7 @@ if __name__ == "__main__":
     parser.add_argument("--host", type=str, default="0.0.0.0")
     parser.add_argument("--port", type=int, default=7860)
     parser.add_argument("--controller-url", type=str, default=None)
+    parser.add_argument("--worker-ip", type=str, default=None)
     parser.add_argument("--concurrency-count", type=int, default=10)
     parser.add_argument(
         "--model-list-mode", type=str, default="reload", choices=["once", "reload"]
@@ -841,6 +844,9 @@ if __name__ == "__main__":
     if not args.controller_url:
         raise ValueError("controller-url is required.")
 
+    if not args.worker_ip:
+        args.worker_ip = os.environ.get("WORKER_IP", None)
+
     models = get_model_list()
 
     sd_worker_url = args.sd_worker_url