Update app.py

app.py

@@ -12,7 +12,7 @@ from huggingface_hub import Repository
 from datetime import datetime
 import scipy.ndimage.interpolation as inter
 import easyocr
-from datasets import load_dataset
+from datasets import load_dataset, Image
 from PIL import Image
 from paddleocr import PaddleOCR
 import socket
@@ -133,10 +133,9 @@ def generate_ocr(Method,img):
             text_output = ocr_with_paddle(img)
 
         new_data=img.reshape(img.shape)
-        # print("^^^^^^^^^^^^^^^",new_data)
         imge = Image.fromarray(new_data.astype(np.uint8),'RGB')
         add_csv = [Method,imge,text_output]
-        print("^^^^^",type(imge))
+    
         # with open(DATA_FILE, "a") as f:
             # writer = csv.writer(f)
             # write the data
@@ -146,6 +145,8 @@ def generate_ocr(Method,img):
         print(commit_url)
         try:
             dataset = load_dataset("pragnakalp/OCR-img-to-text", data_files= "ocr_data.csv", streaming=True)
+            dataset = load_dataset("data", split="train")
+            print("()()()()",dataset[0]["image"])
             print(dataset)
         except Exception as e: 
             print("error in loading data",e)