added explain

app.py

@@ -7,7 +7,7 @@ from transformers import pipeline
 from transformers_interpret import ZeroShotClassificationExplainer
 import string, nltk
 
-models = {'en': 'joeddav/xlm-roberta-large-xnli', #'Narsil/deberta-large-mnli-zero-cls', #'facebook/bart-large-mnli', #'microsoft/deberta-xlarge-mnli', # English
+models = {'en': 'facebook/bart-large-mnli', #'Narsil/deberta-large-mnli-zero-cls', #'microsoft/deberta-xlarge-mnli', # English
           #'de': 'Sahajtomar/German_Zeroshot', # German
           #'es': 'Recognai/zeroshot_selectra_medium', # Spanish
           #'it': 'joeddav/xlm-roberta-large-xnli', # Italian
@@ -171,6 +171,7 @@ def sequence_to_classify(sequence, labels):
         
     model_expl = ZeroShotClassificationExplainer(classifier.model, classifier.tokenizer)
     response_expl = model_expl(sequence, label_clean, hypothesis_template="This passage talks about {}.")
+    print(model_expl.predicted_label)
     
     if len(predicted_labels) == 1:
         response_expl = response_expl[model_expl.predicted_label]