mgyigit commited on
Commit
bb27c74
·
verified ·
1 Parent(s): c98b5e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -143,10 +143,10 @@ def run_inference(mode: str, model_name: str, num_molecules: int, seed_num: str,
143
 
144
  generated_molecule_list = inference_drugs.split("\n")[:-1]
145
 
146
- # Randomly select up to 12 molecules for display
147
  rng = random.Random(config.seed)
148
- if len(generated_molecule_list) > 12:
149
- selected_smiles = rng.choices(generated_molecule_list, k=12)
150
  else:
151
  selected_smiles = generated_molecule_list
152
 
 
143
 
144
  generated_molecule_list = inference_drugs.split("\n")[:-1]
145
 
146
+ # Randomly select up to 9 molecules for display
147
  rng = random.Random(config.seed)
148
+ if len(generated_molecule_list) > 9:
149
+ selected_smiles = rng.choices(generated_molecule_list, k=9)
150
  else:
151
  selected_smiles = generated_molecule_list
152