add zeroshot model

downloadModel.py

@@ -1,13 +1,25 @@
 from transformers import AutoModelForSequenceClassification,AutoTokenizer
 
-#***********************************************************************************************************
-###FOR SENTIMENT CLASSIFICATION
-chkpt='distilbert-base-uncased-finetuned-sst-2-english'
+# ##***********************************************************************************************************##
+# ###FOR SENTIMENT CLASSIFICATION
+# chkpt='distilbert-base-uncased-finetuned-sst-2-english'
+#
+# model=AutoModelForSequenceClassification.from_pretrained(chkpt)
+# tokenizer=AutoTokenizer.from_pretrained(chkpt)
+#
+# #save the pertrained model
+# model.save_pretrained('sentiment_classifier/')
+# tokenizer.save_pretrained('sentiment_classifier/')
+# ##***********************************************************************************************************##
 
-model=AutoModelForSequenceClassification.from_pretrained(chkpt)
-tokenizer=AutoTokenizer.from_pretrained(chkpt)
-
-#save the pertrained model
-model.save_pretrained('sentiment_classifier/')
-tokenizer.save_pretrained('sentiment_classifier/')
-#***********************************************************************************************************
+# # ##***********************************************************************************************************##
+# # ###FOR Zero Shot Text Classification
+# chkpt='valhalla/distilbart-mnli-12-1'
+#
+# model=AutoModelForSequenceClassification.from_pretrained(chkpt)
+# tokenizer=AutoTokenizer.from_pretrained(chkpt)
+#
+# #save the pertrained model
+# model.save_pretrained('zero_shot_clf/')
+# tokenizer.save_pretrained('zero_shot_clf/')
+# # ##***********************************************************************************************************##