thak123 commited on
Commit
70e3867
1 Parent(s): 6c86820

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +6 -4
config.py CHANGED
@@ -8,11 +8,13 @@ VALID_BATCH_SIZE = 4
8
  EPOCHS = 5
9
 
10
  # Folder to contain all the datasets
 
 
11
 
12
- import requests
13
- URL = "https://huggingface.co/thak123/bert-emoji-latvian-twitter-classifier/resolve/main/model.bin"
14
- response = requests.get(URL)
15
- open("model.bin", "wb").write(response.content)
16
 
17
  DATASET_LOCATION = "" #
18
  MODEL_PATH = "model.bin"
 
8
  EPOCHS = 5
9
 
10
  # Folder to contain all the datasets
11
+ from huggingface_hub import hf_hub_download
12
+ hf_hub_download(repo_id="thak123/bert-emoji-latvian-twitter-classifier", filename="model.bin",local_dir="./")
13
 
14
+ # import requests
15
+ # URL = "https://huggingface.co/thak123/bert-emoji-latvian-twitter-classifier/resolve/main/model.bin"
16
+ # response = requests.get(URL)
17
+ # open("model.bin", "wb").write(response.content)
18
 
19
  DATASET_LOCATION = "" #
20
  MODEL_PATH = "model.bin"