Update app.py

app.py

@@ -13,10 +13,10 @@ from datetime import datetime
 import scipy.ndimage.interpolation as inter
 import easyocr
 import datasets
-from datasets import load_dataset, Image, Features, Array3D
+from datasets import load_dataset, Image
 from PIL import Image
 from paddleocr import PaddleOCR
-
+from save_data import flag
             
 """
 Paddle OCR
@@ -80,7 +80,8 @@ def generate_ocr(Method,img):
                 text_output = ocr_with_keras(img)
             if Method == 'PaddleOCR':
                 text_output = ocr_with_paddle(img)
-    
+
+            flag(Method,text_output,img)
             return text_output
         else:
             raise gr.Error("Please upload an image!!!!")