.

app.py

@@ -57,8 +57,11 @@ surface = get_grid_points(coordinates)
 for i, atom in enumerate(smiles_mol.GetAtoms()):
     # For each atom, set the property "molAtomMapNumber" to a custom number, let's say, the index of the atom in the molecule
     atom.SetProp("atomNote", str(atom.GetIdx()))
+
+smiles_image = Chem.Draw.MolToImage(smiles_mol)
+
 # display molecule
-st.image(smiles_mol)
+st.image(smiles_image)
 
 
 # Disable future warnings.