improving layout

app.py

@@ -9,6 +9,7 @@ from AtomLenz import *
 #from utils_graph import *
 from Object_Smiles import Objects_Smiles 
 from rdkit import Chem
+from rdkit.Chem import Draw
 #from robust_detection import wandb_config
 from robust_detection import utils
 from robust_detection.models.rcnn import RCNN
@@ -203,7 +204,7 @@ def atomlenz(modelfile):
             if m is None:
                st.write("No valid chemical structure recognized")
             else:
-                img = Chem.Draw.MolToImage(m)
+                img = Draw.MolToImage(m)
                 st.image(img, caption=f"{smiles}", use_column_width=True)
             print(pred)