LitRL-Inference / run.py
c-gohlke's picture
Update Space
2c329fa verified
raw
history blame
672 Bytes
if __name__ == "__main__": # noqa: INP001
import argparse
import uvicorn
from src.app import create_app
parser = argparse.ArgumentParser()
parser.add_argument("--host", type=str, default="0.0.0.0")
parser.add_argument("--port", type=int, default=7860)
parser.add_argument("--log", type=str, default="info")
parser.add_argument("--reload", action="store_true", help="Reload flag")
args = parser.parse_args()
config = uvicorn.Config(
app=create_app(),
host=args.host,
port=args.port,
log_level=args.log,
reload=args.reload,
)
server = uvicorn.Server(config=config)
server.run()