From ebd31b9f60ea93391e3eb7d58c1a6fe0b813c6a9 Mon Sep 17 00:00:00 2001
From: Navan Chauhan <navanchauhan@gmail.com>
Date: Thu, 24 Sep 2020 15:54:57 +0530
Subject: added tensorflow support to docker-compose

---
 Dockerfile | 3 +++
 README.md  | 4 ++--
 config.ini | 2 +-
 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 
-- 
cgit v1.2.3