Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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('
|
73 |
-
result.save(f"
|
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
|