Update gradio_app.py

gradio_app.py

@@ -50,7 +50,7 @@ def function(model_name: str, num_molecules: int, seed_num: int) -> tuple[PIL.Im
     '''
 
     config = model_configs[model_name]
-    config.inference_sample_num = num_molecules
+    config.sample_num = num_molecules
     config.seed = seed_num
 
     inferer = Inference(config)
@@ -69,8 +69,12 @@ def function(model_name: str, num_molecules: int, seed_num: int) -> tuple[PIL.Im
 
     generated_molecule_list = inference_drugs.split("\n")
 
-    rng = random.Random(seed)
-    selected_molecules = rng.choices(generated_molecule_list,k=12)
+    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 = [Chem.MolFromSmiles(mol) for mol in selected_molecules]
 
     drawOptions = Draw.rdMolDraw2D.MolDrawOptions()