Update app.py

app.py

@@ -30,16 +30,16 @@ 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)
 # DATA_FILENAME2 = "ocr_image"
-DATA_FILE2 = os.mkdir("ocr_data")
+# DATA_FILE2 = os.mkdir("ocr_data")
 HF_TOKEN = os.environ.get("HF_TOKEN")
 DATASET_REPO_ID = "pragnakalp/OCR-img-to-text"
 print("is none?", HF_TOKEN is None)
 try:
     hf_hub_download(
         repo_id=DATASET_REPO_ID,
-        filename=DATA_FILENAME2,
+        filename=DATA_FILENAME,
         cache_dir=DATA_DIRNAME,
-        force_filename=DATA_FILENAME2
+        force_filename=DATA_FILENAME
     )
     
 except:
@@ -136,11 +136,11 @@ def generate_ocr(Method,img):
         imge = Image.fromarray(new_data.astype(np.uint8),'RGB')
         add_csv = [Method,imge,text_output]
     
-        # with open(DATA_FILE, "a") as f:
-            # writer = csv.writer(f)
-            # write the data
-            # writer.writerow(add_csv)
-        os.mkdir(DATA_FILE2)
+        with open(DATA_FILE, "a") as f:
+            writer = csv.writer(f)
+            write the data
+            writer.writerow(add_csv)
+        # os.mkdir(DATA_FILE2)
         commit_url = repo.push_to_hub()
         print(commit_url)
         try: