updated Readme to tell about config.ini

Also, moved away from hard-coding tfWorking value

2 files changed, 8 insertions, 4 deletions

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")