Update config.py
Browse files
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"
|