Owos commited on
Commit
f53266b
·
1 Parent(s): 7665310

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -69,8 +69,8 @@ if uploaded_file is not None:
69
  st.image(result, width=500)
70
  buf = BytesIO()
71
  result.save(buf, format="png")
72
- if len(os.listdir('generated_sampples')) <= 10:
73
- result.save(f"generated_sampples/{str(len(os.listdir('generated_sampples')))}.png")
74
  byte_im = buf.getvalue()
75
  run =st.download_button(label="Download Image", data=byte_im, file_name='afrodreams.jpg', mime="image/png")
76
  #keeping the current style by update the weight
 
69
  st.image(result, width=500)
70
  buf = BytesIO()
71
  result.save(buf, format="png")
72
+ if len(os.listdir('generated_samples')) <= 10:
73
+ result.save(f"generated_samples/{str(len(os.listdir('generated_samples')))}.png")
74
  byte_im = buf.getvalue()
75
  run =st.download_button(label="Download Image", data=byte_im, file_name='afrodreams.jpg', mime="image/png")
76
  #keeping the current style by update the weight