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, Image
+from datasets import load_dataset, Image, Features, Array3D
 from PIL import Image
 from paddleocr import PaddleOCR
 import socket
@@ -135,6 +135,7 @@ def generate_ocr(Method,img):
 
         new_data=img.reshape(img.shape)
         imge = Image.fromarray(new_data.astype(np.uint8),'RGB')
+        feature = Image()
         add_csv = [Method,img,text_output]
     
         with open(DATA_FILE, "a") as f:
@@ -143,11 +144,11 @@ def generate_ocr(Method,img):
             writer.writerow(add_csv)
             commit_url = repo.push_to_hub()
             print(commit_url)
-        # try:
-        #     dataset = load_dataset("pragnakalp/OCR-img-to-text", data_files= "ocr_data.csv", streaming=True)
-        #     print(dataset)
-        # except Exception as e: 
-        #     print("error in loading data",e)
+        try:
+            dataset = load_dataset("pragnakalp/OCR-img-to-text", data_files= "ocr_data.csv", streaming=True)
+            print(dataset.features)
+        except Exception as e: 
+            print("error in loading data",e)
 
             
         # with open(DATA_FILE, "a") as csvfile: