thak123 commited on
Commit
4a27de6
1 Parent(s): 8cfc5e5

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +1 -1
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