Update app.py

app.py

@@ -4,7 +4,8 @@ import random
 from rdkit.Chem import Draw
 from rdkit import Chem
 from rdkit.Chem.Draw import IPythonConsole
-
+import io
+from PIL import Image
 
 class DrugGENConfig:
     submodel='CrossLoss'
@@ -158,7 +159,13 @@ if submitted:
         useSVG=False
     )
     print(type(molecule_image))
-    st.image(molecule_image)
+    png_data = io.BytesIO()
+    molecule_image.save(png_data, format='PNG')
+    png_data.seek(0)
+    
+    # Step 2: Read the PNG image data as a PIL image
+    pil_image = Image.open(png_data)
+    st.image(pil_image)
     
 
 else: