thak123 commited on
Commit
c0bbdb0
·
1 Parent(s): a5b267d

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +1 -3
config.py CHANGED
@@ -9,15 +9,13 @@ EPOCHS = 5
9
 
10
  # Folder to contain all the datasets
11
  DATASET_LOCATION = "" #
12
- MODEL_PATH = "/mnt/data/group3/gaurish/SentimentAnalyserLVTwitter/bert-sentiment/src/trained_models/mbert-7epoch-lower/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
- # 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"