cafierom commited on
Commit
2a27121
·
verified ·
1 Parent(s): af676d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -162,8 +162,9 @@ def similars_node(state: State) -> State:
162
  pic = img.data
163
 
164
  filename = "Similars_image"
165
- with open(filename+".png",'wb+') as outf:
166
- outf.write(pic)
 
167
  # except:
168
  # props_string = 'Could not find similar molecules'
169
  # filename = None
 
162
  pic = img.data
163
 
164
  filename = "Similars_image"
165
+ # with open(filename+".png",'wb+') as outf:
166
+ # outf.write(pic)
167
+ img.save(filename)
168
  # except:
169
  # props_string = 'Could not find similar molecules'
170
  # filename = None