aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorNavan Chauhan <navanchauhan@gmail.com>2020-09-24 15:54:57 +0530
committerNavan Chauhan <navanchauhan@gmail.com>2020-09-24 15:54:57 +0530
commitebd31b9f60ea93391e3eb7d58c1a6fe0b813c6a9 (patch)
tree3dddfa3fec8cf8444cb967ef25f6335941f59ce2
parent6255d82fa7f35d4087e870b5df8f58ecdae1b17e (diff)
added tensorflow support to docker-compose
-rw-r--r--Dockerfile3
-rw-r--r--README.md4
-rw-r--r--config.ini2
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