Update gradcam.py

gradcam.py

@@ -77,7 +77,7 @@ class GradCam():
         topIdx = logits.cpu()[0, :].detach().numpy().argsort()[-1]
         topClass = model.config.id2label[topIdx]
         topScore = probabilities[0][topIdx].item()
-        return {"label": topClass, "score": topScore}
+        return [{"label": topClass, "score": topScore}]
 
     def reshape_transform_vit_huggingface(self, x):
         activations = x[:, 1:, :]