def get_logits(outputs: "ImageClassifierOutput") -> torch.Tensor:
            """
            Gets the logits attribute from ImageClassifierOutput

            :param outputs: outputs of type ImageClassifierOutput from a Huggingface
            :return: model's logit predictions.
            """
            if isinstance(outputs, torch.Tensor):
                return outputs
            return outputs.logits

            Gets the logits attribute from ImageClassifierOutput

            :param outputs: outputs of type ImageClassifierOutput from a Huggingface
            :return: model's logit predictions.
            
art/estimators/classification/hugging_face.py