diff --git a/Dockerfile b/Dockerfile index 22fa24b..b1d894a 100644 --- a/Dockerfile +++ b/Dockerfile @@ -8,7 +8,7 @@ COPY checkpoint /checkpoint # Make changes to the requirements/app here. # This Dockerfile order allows Docker to cache the checkpoint layer # and improve build times if making changes. -RUN pip3 --no-cache-dir install tensorflow==1.15.2 gpt-2-simple starlette uvicorn ujson +RUN pip3 --no-cache-dir install tensorflow==1.15.2 gpt-2-simple starlette==0.13.8 uvicorn ujson COPY app.py / # Clean up APT when done.