AI/rembg
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/danielgatis/rembg synced 2025-08-16 08:56:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

rembg s (http server), add --host parameter (#571)

Browse Source
This commit is contained in:
David Obdržálek [ΔO] 2024-01-23 01:39:29 +01:00 committed by GitHub
parent be319d3489
commit 70b4c44eee
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@ -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

16
rembg/commands/s_command.py
View File

@ -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)