using a lighter zero-shot-classification model

Models.py

@@ -23,8 +23,8 @@ class Models:
         self.ner_dates = pipeline('ner', model=model, tokenizer=tokenizer, aggregation_strategy="simple")
 
         #Zero Shot Classification
-        self.zero_shot_classifier = pipeline("zero-shot-classification", model='facebook/bart-large-mnli')
-        # self.zero_shot_classifier = pipeline("zero-shot-classification", model='valhalla/distilbart-mnli-12-6')
+        # self.zero_shot_classifier = pipeline("zero-shot-classification", model='facebook/bart-large-mnli')
+        self.zero_shot_classifier = pipeline("zero-shot-classification", model='valhalla/distilbart-mnli-12-6')
 
         # Ner
         tokenizer = AutoTokenizer.from_pretrained("dslim/bert-base-NER")