Yin Fang commited on
Commit
1685117
Β·
1 Parent(s): e6f430b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -101,13 +101,13 @@ def opt_process(opt_input):
101
  smis_input = sf.decoder(input)
102
  mols_input = []
103
  mol = Chem.MolFromSmiles(smis_input)
104
- mols_input.append(mol)
105
 
106
  opt_input_img = Draw.MolsToGridImage(
107
- mols_input,
108
  molsPerRow=4,
109
  subImgSize=(200,200),
110
- legends=['' for x in mols]
111
  )
112
 
113
  sf_input = tokenizer(input, return_tensors="pt")
 
101
  smis_input = sf.decoder(input)
102
  mols_input = []
103
  mol = Chem.MolFromSmiles(smis_input)
104
+ mol_input.append(mol)
105
 
106
  opt_input_img = Draw.MolsToGridImage(
107
+ mol_input,
108
  molsPerRow=4,
109
  subImgSize=(200,200),
110
+ legends=['' for x in mol_input]
111
  )
112
 
113
  sf_input = tokenizer(input, return_tensors="pt")