osbm commited on
Commit
808014b
·
1 Parent(s): 67206bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -156,17 +156,17 @@ if submitted:
156
  subImgSize=(250, 250),
157
  maxMols=len(selected_molecules),
158
  # legends=None,
159
- useSVG=False
160
  )
161
  print(type(molecule_image))
162
- png_data = io.BytesIO()
163
- molecule_image.save(png_data, format='PNG')
164
- png_data.seek(0)
165
 
166
  # Step 2: Read the PNG image data as a PIL image
167
- pil_image = Image.open(png_data)
168
- st.image(pil_image)
169
-
170
 
171
  else:
172
  st.warning("Please select a model to make inference")
 
156
  subImgSize=(250, 250),
157
  maxMols=len(selected_molecules),
158
  # legends=None,
159
+ useSVG=True
160
  )
161
  print(type(molecule_image))
162
+ # png_data = io.BytesIO()
163
+ # molecule_image.save(png_data, format='PNG')
164
+ # png_data.seek(0)
165
 
166
  # Step 2: Read the PNG image data as a PIL image
167
+ # pil_image = Image.open(png_data)
168
+ # st.image(pil_image)
169
+ st.image(molecule_image)
170
 
171
  else:
172
  st.warning("Please select a model to make inference")