diff options
Diffstat (limited to 'app/views.py')
| -rw-r--r-- | app/views.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/app/views.py b/app/views.py index 2532736..6af1ee6 100644 --- a/app/views.py +++ b/app/views.py @@ -110,6 +110,7 @@ def wtform(): flash_errors(myform) return render_template('wtform.html', form=myform) +tfWorking = -1 try: from lstm_chem.utils.config import process_config @@ -120,8 +121,11 @@ try: gen = LSTMChemGenerator(modeler) print("Testing Model") gen.sample(1) + tfWorking = 1 except: print("Could not load tensorflow model :/") + tfWorking = 0 + @app.route('/Generate', methods=['GET','POST']) def generate(): |