hard code sizes for convenience

kornia_aug.py

@@ -35,7 +35,7 @@ else:
     im = Image.open("./images/pretty_bird.jpg")
     image = F.pil_to_tensor(im).float() / 255
     ims.append(image) 
-scaler = int(ims[0].shape[0] / 2)
+scaler = 256
 st.sidebar.image(im, caption="Input Image", width=256)