Update app.py

app.py

@@ -6,6 +6,8 @@ import random
 import numpy as np
 #import cv2
 from  PIL import Image, ImageEnhance
+from io import BytesIO
+
 
 st.set_page_config(layout="wide")
 #Create two columns with different width
@@ -56,9 +58,12 @@ if uploaded_file is not None:
         with col2:
             result = Image.open('out.png')
             st.image(result, width=300)
-            run =st.download_button(label="Download Image", data=result.tobytes(), file_name='afrodreams.png', mime="image/png")
+            buf = BytesIO()
+            result.save(buf, format="JPEG")
+            byte_im = buf.getvalue()
+            run =st.download_button(label="Download Image", data=byte_im, file_name='afrodreams.png', mime="image/png")
+
 
 
 
-#img = neural_style.transfer(params)