Update save_data.py

save_data.py

@@ -17,6 +17,17 @@ 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
+    )
+    
+except:
+    print("file not found")
+    
 repo = Repository(
     local_dir="ocr_data", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
 )