kaushikbar commited on
Commit
3692b96
1 Parent(s): 731bd76

using facebook bart large mnli instead of deberta large mnli for english

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import fasttext
6
  from transformers import pipeline
7
 
8
  models = {
9
- 'en': 'Narsil/deberta-large-mnli-zero-cls', # English
10
  #'es': 'Recognai/zeroshot_selectra_medium', # Spanish
11
  #'it': 'joeddav/xlm-roberta-large-xnli', # Italian
12
  #'ru': 'DeepPavlov/xlm-roberta-large-en-ru-mnli', # Russian
 
6
  from transformers import pipeline
7
 
8
  models = {
9
+ 'en': 'facebook/bart-large-mnli', #'Narsil/deberta-large-mnli-zero-cls', # English
10
  #'es': 'Recognai/zeroshot_selectra_medium', # Spanish
11
  #'it': 'joeddav/xlm-roberta-large-xnli', # Italian
12
  #'ru': 'DeepPavlov/xlm-roberta-large-en-ru-mnli', # Russian