Update app.py

app.py

@@ -3,6 +3,8 @@ from trainer import Trainer
 import random
 from rdkit.Chem import Draw
 from rdkit import Chem
+from rdkit.Chem.Draw import IPythonConsole
+
 
 class DrugGENConfig:
     submodel='CrossLoss'