correction to add back environment variable support <3 docker

2023-06-11 01:11:24 +01:00 · 2023-06-11 01:11:24 +01:00 · 3129a0e7e5
commit 3129a0e7e5
parent 3ea31930e5
1 changed files with 1 additions and 1 deletions
--- a/llama_cpp/server/main.py
+++ b/llama_cpp/server/main.py
@ -46,5 +46,5 @@ if __name__ == "__main__":
    app = create_app(settings=settings)

    uvicorn.run(
-        app, host=settings.host, port=settings.port
+        app, host=os.getenv("HOST", settings.host), port=int(os.getenv("PORT", settings.port))
    )