moldenhof commited on
Commit
53d17a4
·
1 Parent(s): 4fd163f

implementing predict smiles

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -205,11 +205,12 @@ if image_file is not None:
205
  molecule['stereo_atoms'] = stereo_atoms
206
  molecule['charge_atoms'] = charge_atoms
207
  mol_graphs.append(molecule)
208
- base_path = pathlib.Path(args.data_path)
209
- image_dir = base_path.joinpath("images")
210
- smiles_dir = base_path.joinpath("smiles")
211
- impath = image_dir.joinpath(f"{image_idx}.png")
212
- smilespath = smiles_dir.joinpath(f"{image_idx}.txt")
 
213
  save_mol_to_file(molecule,'molfile')
214
  mol = Chem.MolFromMolFile('molfile',sanitize=False)
215
  problematic = 0
 
205
  molecule['stereo_atoms'] = stereo_atoms
206
  molecule['charge_atoms'] = charge_atoms
207
  mol_graphs.append(molecule)
208
+ #base_path="./"
209
+ #base_path = pathlib.Path(args.data_path)
210
+ #image_dir = base_path.joinpath("images")
211
+ #smiles_dir = base_path.joinpath("smiles")
212
+ #impath = image_dir.joinpath(f"{image_idx}.png")
213
+ #smilespath = smiles_dir.joinpath(f"{image_idx}.txt")
214
  save_mol_to_file(molecule,'molfile')
215
  mol = Chem.MolFromMolFile('molfile',sanitize=False)
216
  problematic = 0