Update gradio_app.py

gradio_app.py

@@ -51,7 +51,7 @@ def function(model_name: str, num_molecules: int, seed_num: int) -> tuple[PIL.Im
 
     config = model_configs[model_name]
     config.sample_num = num_molecules
-    config.seed = seed_num
+    config.seed = random.randint(0, 10000)
 
     inferer = Inference(config)
     scores = inferer.inference() # create scores_df out of this
@@ -67,13 +67,13 @@ def function(model_name: str, num_molecules: int, seed_num: int) -> tuple[PIL.Im
     with open(new_path) as f:
         inference_drugs = f.read()
 
-    generated_molecule_list = inference_drugs.split("\n")
+    generated_molecule_list = inference_drugs.split("\n")[:-1]
 
     rng = random.Random(config.seed)
     if num_molecules > 12:
         selected_molecules = rng.choices(generated_molecule_list, k=12)
     else:
-        selected_molecules = rng.choices(generated_molecule_list, k=num_molecules)
+        selected_molecules = generated_molecule_list
     
     selected_molecules = [Chem.MolFromSmiles(mol) for mol in selected_molecules if Chem.MolFromSmiles(mol) is not None]
 
@@ -126,12 +126,10 @@ with gr.Blocks() as demo:
                 maximum=10_000,
             )
 
-            val = random.randint(0, 10000)
             seed_num = gr.Number(
                 label="RNG seed value (can be used for reproducibility):",
                 precision=0, # integer input
                 minimum=0,
-                value=val,
             )
 
             submit_button = gr.Button(