diff --git a/Dockerfile b/Dockerfile index 6e793c2..39acfc8 100644 --- a/Dockerfile +++ b/Dockerfile @@ -44,4 +44,7 @@ COPY app /curie-web/app COPY run.py /curie-web COPY api.py /curie-web + +COPY lstm_chem /curie-web/lstm_chem + CMD gunicorn -w 4 api:app -k uvicorn.workers.UvicornWorker -b "0.0.0.0:8080" diff --git a/README.md b/README.md index 005c12d..b0c577c 100644 --- a/README.md +++ b/README.md @@ -10,7 +10,7 @@ Tested on: # Quick Start (Docker-Compose) -You can quickly get started and test Curie-Web without needing to manage dependencies by using the `docker-compose` image. This has all features except AR/3D Model support and LSTM Generator. Otherwise, it is production ready. +You can quickly get started and test Curie-Web without needing to manage dependencies by using the `docker-compose` image. This has all features except AR/3D Model support. Otherwise, it is production ready. ``` @@ -189,4 +189,4 @@ python3 removeSample.py Currently the base model has been trained on 1.5 million compounds and took ~4 hours to train on Google Colab's GPU runtime. -*Information to be added soon* \ No newline at end of file +*Information to be added soon* diff --git a/config.ini b/config.ini index c34c947..344c189 100644 --- a/config.ini +++ b/config.ini @@ -22,4 +22,4 @@ LOG_FOLDER = ./app/logs/ INSTANT = True [FEATURES] -LSTM = False +LSTM = True