allow uploading a batch of images

kornia_aug.py

@@ -28,17 +28,16 @@ uploaded_files = st.sidebar.file_uploader("Choose a file",accept_multiple_files=
 if uploaded_files is not None:
     for uploaded_file in uploaded_files: 
         im = Image.open(uploaded_file)
+        st.sidebar.image(im, caption="Input Image", width=256)
         im=im.resize((512,512))
         image = F.pil_to_tensor(im).float() / 255
         ims.append(image) 
 else:
     im = Image.open("./images/pretty_bird.jpg")
+    st.sidebar.image(im, caption="Input Image", width=256)
     image = F.pil_to_tensor(im).float() / 255
     ims.append(image) 
 scaler = 256
-st.sidebar.image(im, caption="Input Image", width=256)
-
-
 
 # batch size is just for show
 batch_size = st.sidebar.slider("batch_size", min_value=4, max_value=16,value=8)
@@ -72,7 +71,7 @@ nn.Sequential(
 """,
     """
 nn.Sequential(
-   kornia.augmentation.RandomErasing(scale=(.4, .8), ratio=(.3, 1/.3), p=1, same_on_batch=True),
+   kornia.augmentation.RandomErasing(scale=(.4, .8), ratio=(.3, 1/.3), p=1, same_on_batch=False),
 )
 #By setting same_on_batch=True you can apply the same transform across the batch
 """,