allow uploading a batch of images

kornia_aug.py

@@ -23,14 +23,20 @@ st.markdown("# Kornia Augmentations Demo")
 st.sidebar.markdown(
     "[Kornia](https://github.com/kornia/kornia) is a *differentiable* computer vision library for PyTorch."
 )
-uploaded_file = st.sidebar.file_uploader("Choose a file")
-if uploaded_file is not None:
-    im = Image.open(uploaded_file)
+ims=[]
+uploaded_files = st.sidebar.file_uploader("Choose a file")
+if uploaded_files is not None:
+    for uploaded_file in uploaded_files: 
+        im = Image.open(uploaded_file)
+        image = F.pil_to_tensor(im).float() / 255
+        ims.append(image) 
 else:
     im = Image.open("./images/pretty_bird.jpg")
-scaler = int(im.height / 2)
+    image = F.pil_to_tensor(im).float() / 255
+    ims.append(image) 
+scaler = int(ims[0].height / 2)
 st.sidebar.image(im, caption="Input Image", width=256)
-image = F.pil_to_tensor(im).float() / 255
+
 
 
 # batch size is just for show
@@ -110,7 +116,10 @@ if content:
 process = st.button("Next Batch")
 
 # Fake dataloader
-image_batch = torch.stack(batch_size * [image])
+if len(ims)>1:
+    image_batch = torch.stack(ims)
+else:
+    image_batch = torch.stack(batch_size * ims)
 
 
 image_batch.to(device)