diff options
Diffstat (limited to 'app')
-rw-r--r-- | app/__init__.py | 6 | ||||
-rw-r--r-- | app/views.py | 6 |
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") |