from io import BytesIO
buf = BytesIO()
img.save(buf, format="JPEG")
byte_im = buf.getvalue()
Now you can use the st.download_button
btn = col.download_button(
label="Download Image",
data=byte_im,
file_name="imagename.png",
mime="image/jpeg",
)