Update save_data.py

save_data.py

@@ -1,4 +1,5 @@
 import os
+import numpy as np
 import json
 import shutil
 import requests
@@ -6,11 +7,14 @@ import re as r
 from urllib.request import urlopen
 from datetime import datetime
 from datasets import Image
+from PIL import Image
 from huggingface_hub import Repository, upload_file
 
 HF_TOKEN = os.environ.get("HF_TOKEN")
 DATASET_NAME = "OCR-img-to-text"
-DATASET_REPO_URL = f"https://huggingface.co/datasets/pragnakalp/{DATASET_NAME}"
+DATASET_REPO_URL = "https://huggingface.co/datasets/pragnakalp/OCR-img-to-text"
+DATA_FILENAME = "ocr_data.csv"
+DATA_FILE = os.path.join("ocr_data", DATA_FILENAME)
 DATASET_REPO_ID = "pragnakalp/OCR-img-to-text"
 print("is none?", HF_TOKEN is None)
 REPOSITORY_DIR = "data"