navan
/
Curie-Web
mirror of https://github.com/navanchauhan/Curie-Web.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

added tf import strategy

This commit is contained in:
Navan Chauhan 2020-08-04 21:39:26 +05:30
parent de479dd421
commit 3340d0f9f2
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
app/views.py
View File

@ -113,6 +113,13 @@ def wtform():
tfWorking = -1 tfWorking = -1
try: try:
import tensorflow as tf
tfWorking = 1
except:
print("Could not load tensorflow model :/")
tfWorking = 0
if tfWorking == 1:
from lstm_chem.utils.config import process_config from lstm_chem.utils.config import process_config
from lstm_chem.model import LSTMChem from lstm_chem.model import LSTMChem
from lstm_chem.generator import LSTMChemGenerator from lstm_chem.generator import LSTMChemGenerator
@ -121,11 +128,6 @@ try:
gen = LSTMChemGenerator(modeler) gen = LSTMChemGenerator(modeler)
print("Testing Model") print("Testing Model")
gen.sample(1) gen.sample(1)
tfWorking = 1
except:
print("Could not load tensorflow model :/")
tfWorking = 0
@app.route('/Generate', methods=['GET','POST']) @app.route('/Generate', methods=['GET','POST'])
def generate(): def generate():
@ -140,7 +142,10 @@ def generate():
if request.method == 'POST' and form.validate_on_submit(): if request.method == 'POST' and form.validate_on_submit():
result = gen.sample(form.n.data) result = gen.sample(form.n.data)
return render_template('generate.html',expName=j["exp_name"],epochs=j["num_epochs"],optimizer=j["optimizer"].capitalize(), form=form,result=result) if tfWorking == 0:
flash("Failed to initialise the model!","danger")
else:
return render_template('generate.html',expName=j["exp_name"],epochs=j["num_epochs"],optimizer=j["optimizer"].capitalize(), form=form,result=result)
return render_template('generate.html',expName=j["exp_name"],epochs=j["num_epochs"],optimizer=j["optimizer"].capitalize(), form=form) return render_template('generate.html',expName=j["exp_name"],epochs=j["num_epochs"],optimizer=j["optimizer"].capitalize(), form=form)