Spaces:

ceyda
/

kornia-augmentations-tester

Runtime error

ceyda commited on Aug 24, 2022

Commit

3682141

1 Parent(s): 6e3a15d

Update kornia_aug.py

Files changed (1) hide show

kornia_aug.py CHANGED Viewed

@@ -34,7 +34,7 @@ else:
     im = Image.open("./images/pretty_bird.jpg")
     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)

     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)
 st.sidebar.image(im, caption="Input Image", width=256)