Update save_data.py

save_data.py

@@ -21,21 +21,23 @@ REPOSITORY_DIR = "data"
 LOCAL_DIR = 'data_local'
 os.makedirs(LOCAL_DIR,exist_ok=True)
 
-# try:
-#     hf_hub_download(
-#         repo_id=DATASET_REPO_ID,
-#         filename=DATA_FILENAME,
-#         cache_dir=DATA_DIRNAME,
-#         force_filename=DATA_FILENAME
-#     )
+try:
+    hf_hub_download(
+        repo_id=DATASET_REPO_ID,
+        filename=DATA_FILENAME,
+        cache_dir=DATA_DIRNAME,
+        force_filename=DATA_FILENAME
+    )
     
-# except:
-#     print("file not found")
+except:
+    print("file not found")
     
-# repo = Repository(
-#     local_dir="ocr_data", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
-# )
-# repo.git_pull()
+try:
+    repo = Repository(local_dir="ocr_data", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN)
+    repo.git_pull()
+except Exception as e:
+    print("Error occurred during git pull:", e)
+
 # repo = Repository(local_dir="ocr_data", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN)
 # repo.git_pull()