NavyaNayer commited on
Commit
9e49caf
·
verified ·
1 Parent(s): 222277e

Delete download_models.py

Browse files
Files changed (1) hide show
  1. download_models.py +0 -19
download_models.py DELETED
@@ -1,19 +0,0 @@
1
- import os
2
- from transformers import AutoTokenizer, AutoModelForSequenceClassification
3
-
4
- def download_base_models():
5
- # Create models directory
6
- os.makedirs("pretrained_models", exist_ok=True)
7
-
8
- print("Downloading BERT base model...")
9
- # Download and save BERT base model
10
- tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
11
- model = AutoModelForSequenceClassification.from_pretrained("bert-base-uncased")
12
-
13
- # Save models locally
14
- tokenizer.save_pretrained("pretrained_models/bert-base-uncased")
15
- model.save_pretrained("pretrained_models/bert-base-uncased")
16
- print("Base models downloaded successfully!")
17
-
18
- if __name__ == "__main__":
19
- download_base_models()