Update gradio_app.py

gradio_app.py

@@ -75,7 +75,7 @@ def function(model_name: str, num_molecules: int, seed_num: int) -> tuple[PIL.Im
     else:
         selected_molecules = rng.choices(generated_molecule_list, k=num_molecules)
     
-    selected_molecules = [Chem.MolFromSmiles(mol) for mol in selected_molecules if not None]
+    selected_molecules = [Chem.MolFromSmiles(mol) for mol in selected_molecules if Chem.MolFromSmiles(mol) is not None]
 
     drawOptions = Draw.rdMolDraw2D.MolDrawOptions()
     drawOptions.prepareMolsBeforeDrawing = False