scd

utilitis.py

@@ -7,11 +7,10 @@ import numpy as np
 import time
 
 model_Hugging_path = "Noureddinesa/Output_LayoutLMv3_v7"
-ocr = PaddleOCR(use_angle_cls=False,
-                lang='fr',
-                  rec=False,
-                  use_tensorrt = True
-                )
+
+def Paddle():
+    ocr = PaddleOCR(use_angle_cls=False,lang='fr',rec=False)
+    return ocr
 
 #############################################################################
 #############################################################################
@@ -39,6 +38,7 @@ def processbbox(BBOX, width, height):
 
 
 def Preprocess(image):
+    ocr = Paddle()
     image_array = np.array(image)
     width, height = image.size
     results = ocr.ocr(image_array,  cls=False,rec = True)
@@ -83,6 +83,7 @@ def drop_null_bbox(dictionary):
         del dictionary[key]
 
 def get_word(bboxes,image):
+    ocr = Paddle()
     x_min, y_min, x_max, y_max = bboxes
     roi = image.crop((x_min, y_min, x_max, y_max)) # Region of intrest
     roi_np = np.array(roi) # To array