Hi @Roshni_Pathak , inputs must be provided as a tensor or tuple of tensors to appropriately compute attributions. If the tensors are contained as values within the dictionary, then you can define a wrapper function which takes the tensors as input and constructs an appropriate dictionary and calls my_model with this dictionary. An example of this approach using a wrapper method with GuidedGradCam can be found in the discussion here 40. Hope this helps!