osbm commited on
Commit
71a7f9f
·
1 Parent(s): 7de09a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -3,6 +3,7 @@ from trainer import Trainer
3
  import random
4
  from rdkit.Chem import Draw
5
  from rdkit import Chem
 
6
 
7
  class DrugGENConfig:
8
  submodel='CrossLoss'
 
3
  import random
4
  from rdkit.Chem import Draw
5
  from rdkit import Chem
6
+ from rdkit.Chem.Draw import IPythonConsole
7
 
8
  class DrugGENConfig:
9
  submodel='CrossLoss'