from transformers_interpret import SequenceClassificationExplainer, MultiLabelClassificationExplainer def explainer(input_statement, model, tokenizer): cls_explainer = SequenceClassificationExplainer(model, tokenizer) # cls_explainer = MultiLabelClassificationExplainer(model, tokenizer) word_attributions = cls_explainer(input_statement) return dict(word_attributions)