Update app.py

app.py

@@ -30,7 +30,7 @@ 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.csv"
-# DATA_FILE2 = os.path.join("ocr_image", DATA_FILENAME2)
+DATA_FILE2 = os.path.join("OCR-img-to-text")
 HF_TOKEN = os.environ.get("HF_TOKEN")
 DATASET_REPO_ID = "pragnakalp/OCR-img-to-text"
 print("is none?", HF_TOKEN is None)
@@ -162,12 +162,12 @@ def generate_ocr(Method,img):
             print(commit_url)
 
         
-        # Image.fromarray(img).save(DATA_FILE2 +"/"+ f"{image_id}.png")
-        # with open(DATA_FILE2, "wt") as csvfile:
-        #     writer = csv.writer(
-        #         csvfile, delimiter=" ", quotechar="|", quoting=csv.QUOTE_MINIMAL
-        #     )
-        #     writer.writerow([0])
+        Image.fromarray(img).save(DATA_FILE2 +"/"+ f"{image_id}.png")
+        with open(DATA_FILE2, "wt") as csvfile:
+            writer = csv.writer(
+                csvfile, delimiter=" ", quotechar="|", quoting=csv.QUOTE_MINIMAL
+            )
+            writer.writerow(['0'])
             
         # try:
         #     dataset = load_dataset("pragnakalp/OCR-img-to-text", streaming=True)