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 |
|