Update Home.py

Home.py

@@ -150,14 +150,13 @@ if uploaded_file is not None:
                 run = ste.download_button(button_text="Download Image", data=byte_im, download_filename='afrodreams.jpg', mime="image/png")
                 
 
-
-
-# selectiuing random iamges to be displayed
-img_names = [os.path.join('generated_samples', img) for img in os.listdir('generated_samples')]
-five_rand_imgs1 = get_random_subset(img_names, 5)
-st.subheader('\n\n\n\n\n\n\n\n Examples of some Generate Images')
-display_random_images(five_rand_imgs1, 'Generate image', size=(20, 15))
-
+                if run==True 
+                    # selectiuing random iamges to be displayed
+                    img_names = [os.path.join('generated_samples', img) for img in os.listdir('generated_samples')]
+                    five_rand_imgs1 = get_random_subset(img_names, 5)
+                    st.subheader('\n\n\n\n\n\n\n\n Examples of some Generate Images')
+                    display_random_images(five_rand_imgs1, 'Generate image', size=(20, 15))
+