Update app.py

app.py

@@ -8,6 +8,9 @@ import numpy as np
 from  PIL import Image, ImageEnhance
 from io import BytesIO
 import streamlit_ext as ste #for download button not to rerun
+import huggingface_hub
+
+HF_TOKEN = os.environ.get("HF_TOKEN")
 
 st.set_page_config(layout="wide")
 #Create two columns with different width
@@ -48,7 +51,7 @@ if uploaded_file is not None:
     params = neural_style.TransferParams()
     params.gpu = "c"
     params.backend = "mkl"
-    params.image_size = 300
+    params.image_size = 400
     params.content_image = uploaded_file
     params.style_weight = style_weight
     keep_style = False
@@ -71,7 +74,14 @@ if uploaded_file is not None:
                 buf = BytesIO()
                 result.save(buf, format="png")
                 if len(os.listdir('generated_samples')) <= 10:
-                    result.save(f"generated_samples/{str(len(os.listdir('generated_samples')))}.png")
+                
+                    _ = huggingface_hub.upload_file(path_or_fileobj = 'out.png',
+                            path_in_repo =f"generated_samples/{str(len(os.listdir('generated_samples')))}.png",
+                            repo_id='AfrodreamsAI/afrodreams',
+                            repo_type='space',
+                            token=HF_TOKEN
+                        ) 
+                    
                 byte_im = buf.getvalue()
                 run =ste.download_button(button_text="Download Image", data=byte_im, download_filename='afrodreams.jpg', mime="image/png")
         #keeping the current style by update the weight