Update app.py

app.py

@@ -156,17 +156,17 @@ if submitted:
         subImgSize=(250, 250),
         maxMols=len(selected_molecules),
         # legends=None,
-        useSVG=False
+        useSVG=True
     )
     print(type(molecule_image))
-    png_data = io.BytesIO()
-    molecule_image.save(png_data, format='PNG')
-    png_data.seek(0)
+    # 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)
-    
+    # pil_image = Image.open(png_data)
+    # st.image(pil_image)
+    st.image(molecule_image)
 
 else:
     st.warning("Please select a model to make inference")