Update app.py

app.py

@@ -132,9 +132,9 @@ def generate_ocr(Method,img):
         if Method == 'PaddleOCR':
             text_output = ocr_with_paddle(img)
 
-        new_data=img.reshape(img.shape)
-        imge = Image.fromarray(new_data.astype(np.uint8),'RGB')
-        add_csv = [Method,imge,text_output]
+        # new_data=img.reshape(img.shape)
+        # imge = Image.fromarray(new_data.astype(np.uint8),'RGB')
+        add_csv = [Method,cv2.imwrite("myimage.jpg", img),text_output]
     
         with open(DATA_FILE, "a") as f:
             writer = csv.writer(f)
@@ -145,8 +145,6 @@ 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("pragnakalp/OCR-img-to-text/data", split="train")
-            print("()()()()",dataset[0]["image"])
             print(dataset)
         except Exception as e: 
             print("error in loading data",e)