Spaces:

probing-vits
/

class-attention-map

Runtime error

sayakpaul HF staff commited on May 3, 2022

Commit

0c48c4a

•

1 Parent(s): b059242

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -25,7 +25,7 @@ def preprocess_image(orig_image: Image, size: int):
         image_resized, (resize_size, resize_size), method="bicubic"
     )
     image_resized = crop_layer(image_resized)
-    return image_resized, norm_layer(image_resized).numpy()
 # Reference:

         image_resized, (resize_size, resize_size), method="bicubic"
     )
     image_resized = crop_layer(image_resized)
+    return image_resized.numpy().squeeze(), norm_layer(image_resized).numpy()
 # Reference: