Update Home.py

Home.py

@@ -7,7 +7,7 @@ import numpy as np
 from  PIL import Image, ImageEnhance
 from io import BytesIO
 import matplotlib.pyplot as plt
-#import streamlit_ext as ste #for download button not to rerun
+import streamlit_ext as ste #for download button not to rerun
 from huggingface_hub import upload_file
 
 HF_TOKEN = os.environ.get("HF_TOKEN")
@@ -31,9 +31,6 @@ with col1:               # To display the header text using css style
     
     
 
-#with st.sidebar.expander("About the App"):
-#     st.write("""
-#       This app takes in your image and styles it with a unique african art.""")
 
 
 #Add file uploader to allow users to upload photos
@@ -64,11 +61,8 @@ def display_random_images(five_rand_imgs, display_type, size= (15, 6)):
     
     st.pyplot(fig)
 
-def select_title_from_check_box(index, names_dict):
-    return names_dict[index]
-def format_func(five_rand_imgs):
-    st.write(' the lenght is', len(five_rand_imgs))
-    return list(range(len(five_rand_imgs)))
+
+
 path = 'stylesv2'
     
     
@@ -142,18 +136,18 @@ if uploaded_file is not None:
                 st.image(result, width=300)
                 buf = BytesIO()
                 result.save(buf, format="png")
-                if len(os.listdir('generated_samples')) <= 10:
-                    img_file_name = f"generated_samples/{str(len(os.listdir('generated_samples')))}.png"
-                
-                    _ = upload_file(path_or_fileobj = 'out.png',
-                            path_in_repo = img_file_name,
-                            repo_id='AfrodreamsAI/afrodreams',
-                            repo_type='space',
-                            token=HF_TOKEN
-                        ) 
+
+                img_file_name = f"generated_samples/{str(len(os.listdir('generated_samples')))}.png"
+            
+                _ = upload_file(path_or_fileobj = 'out.png',
+                        path_in_repo = img_file_name,
+                        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")
+                run = ste.download_button(button_text="Download Image", data=byte_im, download_filename='afrodreams.jpg', mime="image/png")