Update app.py

app.py

@@ -7,7 +7,7 @@ import numpy as np
 #import cv2
 from  PIL import Image, ImageEnhance
 from io import BytesIO
-
+import ste #for download button not to rerun
 
 st.set_page_config(layout="wide")
 #Create two columns with different width
@@ -73,7 +73,7 @@ if uploaded_file is not None:
                 if len(os.listdir('generated_samples')) <= 10:
                     result.save(f"generated_samples/{str(len(os.listdir('generated_samples')))}.png")
                 byte_im = buf.getvalue()
-                run =st.download_button(label="Download Image", data=byte_im, file_name='afrodreams.jpg', mime="image/png")
+                run =ste.download_button(label="Download Image", data=byte_im, file_name='afrodreams.jpg', mime="image/png")
         #keeping the current style by update the weight 
         keep_style = st.sidebar.checkbox("Keep current style")