Update config.py

config.py

@@ -9,22 +9,16 @@ VALID_BATCH_SIZE = 4
 EPOCHS = 5
 
 # Folder to contain all the datasets
-# from huggingface_hub import hf_hub_download
-# print("hi")
-# hf_hub_download(repo_id="thak123/bert-emoji-latvian-twitter-classifier", filename="model.bin",local_dir="./")
+from huggingface_hub import hf_hub_download
 
-# from huggingface_hub import snapshot_download
-# snapshot_download(repo_id="thak123/bert-emoji-latvian-twitter-classifier", allow_patterns="*.bin")
+DATASET_LOCATION = "" # 
 
 
-# import requests
-URL = "https://huggingface.co/thak123/bert-emoji-latvian-twitter-classifier/resolve/main/model.bin"
-response = requests.get(URL)
-open("model.bin", "wb").write(response.content)
+MODEL_PATH = hf_hub_download(repo_id="thak123/bert-emoji-latvian-twitter-classifier", filename="model.bin")
 
+# from huggingface_hub import snapshot_download
+# snapshot_download(repo_id="thak123/bert-emoji-latvian-twitter-classifier", allow_patterns="*.bin")
 
-DATASET_LOCATION = "" # 
-MODEL_PATH = "model.bin"
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')