jannisborn commited on
Commit
bebd86d
·
unverified ·
1 Parent(s): a4eba41
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -33,8 +33,9 @@ def run_inference(
33
  )
34
  model = AdvancedManufacturing(config, target=target_binding_energy)
35
  samples = list(model.sample(number_of_samples))
 
36
 
37
- return draw_grid_generate(samples=samples, n_cols=5, seeds=[])
38
 
39
 
40
  if __name__ == "__main__":
 
33
  )
34
  model = AdvancedManufacturing(config, target=target_binding_energy)
35
  samples = list(model.sample(number_of_samples))
36
+ seeds = [] if primer_smiles == "" else [primer_smiles]
37
 
38
+ return draw_grid_generate(samples=samples, n_cols=5, seeds=seeds)
39
 
40
 
41
  if __name__ == "__main__":