aboutsummaryrefslogtreecommitdiff
path: root/app
diff options
context:
space:
mode:
Diffstat (limited to 'app')
-rw-r--r--app/__init__.py6
-rw-r--r--app/views.py6
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")