Spaces:
Sleeping
Sleeping
Update config.py
Browse files
config.py
CHANGED
@@ -9,15 +9,13 @@ EPOCHS = 5
|
|
9 |
|
10 |
# Folder to contain all the datasets
|
11 |
DATASET_LOCATION = "" #
|
12 |
-
MODEL_PATH = "/
|
13 |
|
14 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
15 |
|
16 |
# MBERT Raw Version
|
17 |
# BERT_PATH = "bert-base-multilingual-cased"
|
18 |
|
19 |
-
# 2 EPOCH Version
|
20 |
-
# BERT_PATH = "bert-twitter-fine-tunning/LatvianTwittermBERT-v1"
|
21 |
|
22 |
# 7 EPOCH Version
|
23 |
BERT_PATH = "FFZG-cleopatra/bert-emoji-latvian-twitter"
|
|
|
9 |
|
10 |
# Folder to contain all the datasets
|
11 |
DATASET_LOCATION = "" #
|
12 |
+
MODEL_PATH = "https://huggingface.co/FFZG-cleopatra/lv-cros-sentimentor/blob/main/model.bin"
|
13 |
|
14 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
15 |
|
16 |
# MBERT Raw Version
|
17 |
# BERT_PATH = "bert-base-multilingual-cased"
|
18 |
|
|
|
|
|
19 |
|
20 |
# 7 EPOCH Version
|
21 |
BERT_PATH = "FFZG-cleopatra/bert-emoji-latvian-twitter"
|