column layout

app.py

@@ -17,7 +17,7 @@ checkpoint_path = "Checkpoints/150epochs.chkpt"
 st.title("Generating Abstract Art")
 
 st.sidebar.subheader("Configurations")
-seed = st.sidebar.slider('Seed', -10000, 10000, 0)
+seed = st.sidebar.slider('Seed', -100000, 100000, 0)
 
 num_images = st.sidebar.slider('Number of Images', 1, 10, 1)
 
@@ -27,7 +27,7 @@ use_srgan = st.sidebar.selectbox(
 )
 
 generate = st.sidebar.button("Generate")
-
+st.write("Get started using the left side bar :sunglasses:")
 
 # caching the expensive model loading 
 
@@ -65,16 +65,19 @@ if generate:
             enhanced_fakes = esrgan_generator(fakes).detach().cpu()
         color_match = color_histogram_mapping(enhanced_fakes, fakes.cpu())
 
+        cols = st.columns(num_images)
         for i in range(len(color_match)):
             # denormalize and permute to correct color channel
-            st.image(denormalize_images(color_match[i]).permute(1, 2, 0).numpy())
+            cols[i].image(denormalize_images(color_match[i]).permute(1, 2, 0).numpy(), use_column_width=True)
 
 
     # default setting -> vanilla dcgan generation
     if use_srgan == "No":
         fakes = fakes.cpu()
+
+        cols = st.columns(num_images)
         for i in range(len(fakes)):
-            st.image(denormalize_images(fakes[i]).permute(1, 2, 0).numpy())
+            cols[i].image(denormalize_images(fakes[i]).permute(1, 2, 0).numpy(), use_column_width=True)