Spaces:
Runtime error
Runtime error
kaushikbar
commited on
Commit
•
f5b762d
1
Parent(s):
31d359c
added explain
Browse files
app.py
CHANGED
@@ -15,12 +15,12 @@ models = {'en': 'joeddav/xlm-roberta-large-xnli', #'Narsil/deberta-large-mnli-ze
|
|
15 |
#'tr': 'vicgalle/xlm-roberta-large-xnli-anli', # Turkish
|
16 |
'no': 'NbAiLab/nb-bert-base-mnli'} # Norsk
|
17 |
|
18 |
-
hypothesis_templates = {'en': 'This
|
19 |
-
'de': 'Dieses beispiel ist {}.', # German
|
20 |
-
'es': 'Este ejemplo es {}.', # Spanish
|
21 |
-
'it': 'Questo esempio è {}.', # Italian
|
22 |
-
'ru': 'Этот пример {}.', # Russian
|
23 |
-
'tr': 'Bu örnek {}.', # Turkish
|
24 |
'no': 'Dette eksempelet er {}.'} # Norsk
|
25 |
|
26 |
classifiers = {'en': pipeline("zero-shot-classification", hypothesis_template=hypothesis_templates['en'],
|
@@ -170,7 +170,7 @@ def sequence_to_classify(sequence, labels):
|
|
170 |
puncts = list(string.punctuation)
|
171 |
|
172 |
model_expl = ZeroShotClassificationExplainer(classifier.model, classifier.tokenizer)
|
173 |
-
response_expl = model_expl(sequence, label_clean, hypothesis_template="This
|
174 |
|
175 |
if len(predicted_labels) == 1:
|
176 |
response_expl = response_expl[model_expl.predicted_label]
|
|
|
15 |
#'tr': 'vicgalle/xlm-roberta-large-xnli-anli', # Turkish
|
16 |
'no': 'NbAiLab/nb-bert-base-mnli'} # Norsk
|
17 |
|
18 |
+
hypothesis_templates = {'en': 'This passage talks about {}.', # English
|
19 |
+
#'de': 'Dieses beispiel ist {}.', # German
|
20 |
+
#'es': 'Este ejemplo es {}.', # Spanish
|
21 |
+
#'it': 'Questo esempio è {}.', # Italian
|
22 |
+
#'ru': 'Этот пример {}.', # Russian
|
23 |
+
#'tr': 'Bu örnek {}.', # Turkish
|
24 |
'no': 'Dette eksempelet er {}.'} # Norsk
|
25 |
|
26 |
classifiers = {'en': pipeline("zero-shot-classification", hypothesis_template=hypothesis_templates['en'],
|
|
|
170 |
puncts = list(string.punctuation)
|
171 |
|
172 |
model_expl = ZeroShotClassificationExplainer(classifier.model, classifier.tokenizer)
|
173 |
+
response_expl = model_expl(sequence, label_clean, hypothesis_template="This passage talks about {}.")
|
174 |
|
175 |
if len(predicted_labels) == 1:
|
176 |
response_expl = response_expl[model_expl.predicted_label]
|