Update config.py
Browse files
config.py
CHANGED
@@ -9,7 +9,7 @@ EPOCHS = 5
|
|
9 |
|
10 |
# Folder to contain all the datasets
|
11 |
DATASET_LOCATION = "" #
|
12 |
-
MODEL_PATH = "thak123/bert-emoji-latvian-twitter-classifier"
|
13 |
|
14 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
15 |
|
|
|
9 |
|
10 |
# Folder to contain all the datasets
|
11 |
DATASET_LOCATION = "" #
|
12 |
+
MODEL_PATH = "https://huggingface.co/thak123/bert-emoji-latvian-twitter-classifier/resolve/main/model.bin"
|
13 |
|
14 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
15 |
|