diff --git a/model_server/scripts/run_server.py b/model_server/scripts/run_server.py index ca8126e4de0c782009373e078d70a637fbb70b2a..b40ff2b399008f26f24d1d0ab5bf205650b03c95 100644 --- a/model_server/scripts/run_server.py +++ b/model_server/scripts/run_server.py @@ -30,12 +30,12 @@ def parse_args(): return parser.parse_args() -def main(args) -> None: +def main(args, app_name='model_server.base.api:app') -> None: print('CLI args:\n' + str(args)) server_process = Process( target=uvicorn.run, - args=('model_server.base.api:app',), + args=(app_name,), kwargs={ 'app_dir': '.', 'host': args.host,