download script for models

downloadModel.py

@@ -0,0 +1,13 @@
+from transformers import AutoModelForSequenceClassification,AutoTokenizer
+
+#***********************************************************************************************************
+###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/')
+#***********************************************************************************************************