ImageClassificationExplainer: AttributeError: ndim when trying to visualize. about transformers-interpret HOT 3 OPEN

I have the same problem when playing with the exemplary code

`from transformers import AutoFeatureExtractor, AutoModelForImageClassification
from transformers_interpret import ImageClassificationExplainer
from PIL import Image
import requests

model_name = "google/vit-base-patch16-224"
model = AutoModelForImageClassification.from_pretrained(model_name)
feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)

image_link = "https://imagesvc.meredithcorp.io/v3/mm/image?url=https%3A%2F%2Fstatic.onecms.io%2Fwp-content%2Fuploads%2Fsites%2F47%2F2020%2F08%2F16%2Fgolden-retriever-177213599-2000.jpg"

image = Image.open(requests.get(image_link, stream=True).raw)

image_classification_explainer = ImageClassificationExplainer(model=model, feature_extractor=feature_extractor)

image_attributions = image_classification_explainer(
image
)

print(image_attributions.shape)

image_classification_explainer.visualize(
method="heatmap",
side_by_side=True,
outlier_threshold=0.03

)`

from transformers-interpret.

Do you know the solution to this error?

from transformers-interpret.