From 70b4c44eeee7da97cd163185f6f39fc26f959781 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?David=20Obdr=C5=BE=C3=A1lek=20=5B=CE=94O=5D?= Date: Tue, 23 Jan 2024 01:39:29 +0100 Subject: [PATCH] rembg s (http server), add --host parameter (#571) --- README.md | 4 ++++ rembg/commands/s_command.py | 16 ++++++++++++---- 2 files changed, 16 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index 24113c5..91bb84d 100644 --- a/README.md +++ b/README.md @@ -190,6 +190,10 @@ rembg p -w path/to/input path/to/output Used to start http server. +``` +rembg s --host 0.0.0.0 --port 5000 --log_level info +``` + To see the complete endpoints documentation, go to: `http://localhost:5000/api`. Remove the background from an image url diff --git a/rembg/commands/s_command.py b/rembg/commands/s_command.py index bacc490..b7a484e 100644 --- a/rembg/commands/s_command.py +++ b/rembg/commands/s_command.py @@ -31,6 +31,14 @@ from ..sessions.base import BaseSession show_default=True, help="port", ) +@click.option( + "-h", + "--host", + default="0.0.0.0", + type=str, + show_default=True, + help="host", +) @click.option( "-l", "--log_level", @@ -47,7 +55,7 @@ from ..sessions.base import BaseSession show_default=True, help="number of worker threads", ) -def s_command(port: int, log_level: str, threads: int) -> None: +def s_command(port: int, host: str, log_level: str, threads: int) -> None: """ Command-line interface for running the FastAPI web server. @@ -282,7 +290,7 @@ def s_command(port: int, log_level: str, threads: int) -> None: app = gr.mount_gradio_app(app, interface, path="/") return app - print(f"To access the API documentation, go to http://localhost:{port}/api") - print(f"To access the UI, go to http://localhost:{port}") + print(f"To access the API documentation, go to http://{'localhost' if host == '0.0.0.0' else host}:{port}/api") + print(f"To access the UI, go to http://{'localhost' if host == '0.0.0.0' else host}:{port}") - uvicorn.run(gr_app(app), host="0.0.0.0", port=port, log_level=log_level) + uvicorn.run(gr_app(app), host=host, port=port, log_level=log_level)