added tensorflow support to docker-compose

3 files changed, 6 insertions, 3 deletions

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