Spaces:
Running
Running
Update app.py
Browse files
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
|
147 |
rng = random.Random(config.seed)
|
148 |
-
if len(generated_molecule_list) >
|
149 |
-
selected_smiles = rng.choices(generated_molecule_list, k=
|
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 |
|