From 4b48546cea6c894d735f6c9e9d271f4f6a15a57e Mon Sep 17 00:00:00 2001
From: Navan Chauhan <navanchauhan@gmail.com>
Date: Thu, 10 Sep 2020 12:25:30 +0530
Subject: updated Readme to tell about config.ini

Also, moved away from hard-coding tfWorking value
---
 app/__init__.py | 6 ++++++
 app/views.py    | 6 ++----
 2 files changed, 8 insertions(+), 4 deletions(-)

(limited to 'app')

diff --git a/app/__init__.py b/app/__init__.py
index b405e40..1b2171c 100644
--- a/app/__init__.py
+++ b/app/__init__.py
@@ -12,6 +12,12 @@ DB_NAME = config['DATABASE']['NAME']
 UPLOAD_FOLDER = config['FILES']['UPLOAD_FOLDER']
 LOG_FOLDER = config['FILES']['LOG_FOLDER']
 INSTANT_EXEC = config['EXECUTION']['INSTANT']
+LSTM = config['FEATURES']['LSTM']
+
+if LSTM == 'True':
+    LSTM = True
+else:
+    LSTM = False
 
 if INSTANT_EXEC == 'True':
     INSTANT_EXEC = True
diff --git a/app/views.py b/app/views.py
index c296f68..c272e73 100644
--- a/app/views.py
+++ b/app/views.py
@@ -247,11 +247,9 @@ def generate_pdbqts():
     return render_template('pdbqt_form.html',form=myform)
             
 
-tfWorking = 0
-
-if tfWorking == -1:
+if app.config['LSTM']:
     try:
-        import tensorrflow as tf
+        import tensorflow as tf
         tfWorking = 1
     except Exception as e:
         log(e,"EXCEPTION")
-- 
cgit v1.2.3