Update utility/utils.py

utility/utils.py

@@ -21,8 +21,9 @@ logging.basicConfig(
 )
 
 # Set the PaddleOCR home directory to a writable location
+import os
 
-os.environ['PADDLEOCR_HOME'] = '/tmp/.paddleocr' 
+os.environ["PADDLEOCR_HOME"] = "/app/paddleocr_models"
 
 RESULT_FOLDER = 'static/results/'
 JSON_FOLDER = 'static/json/'
@@ -107,14 +108,22 @@ def ocr_with_paddle(img):
     boxes = []
 
     # Initialize PaddleOCR
+    # ocr = PaddleOCR(
+    #     lang='en', 
+    #     use_angle_cls=True,
+    #     det_model_dir=os.path.join(os.environ['PADDLEOCR_HOME'], 'whl/det'),
+    #     rec_model_dir=os.path.join(os.environ['PADDLEOCR_HOME'], 'whl/rec/en/en_PP-OCRv4_rec_infer'),
+    #     cls_model_dir=os.path.join(os.environ['PADDLEOCR_HOME'], 'whl/cls/ch_ppocr_mobile_v2.0_cls_infer')
+    # )
     ocr = PaddleOCR(
-        lang='en', 
         use_angle_cls=True,
-        det_model_dir=os.path.join(os.environ['PADDLEOCR_HOME'], 'whl/det'),
-        rec_model_dir=os.path.join(os.environ['PADDLEOCR_HOME'], 'whl/rec/en/en_PP-OCRv4_rec_infer'),
-        cls_model_dir=os.path.join(os.environ['PADDLEOCR_HOME'], 'whl/cls/ch_ppocr_mobile_v2.0_cls_infer')
+        lang='en',
+        det_model_dir='/app/paddleocr_models/whl/det/ch_ppocr_mobile_v2.0_det_infer',
+        rec_model_dir='/app/paddleocr_models/whl/rec/ch_ppocr_mobile_v2.0_rec_infer',
+        cls_model_dir='/app/paddleocr_models/whl/cls/ch_ppocr_mobile_v2.0_cls_infer'
     )
 
+
     # Check if img is a file path or an image array
     if isinstance(img, str):
         img = cv2.imread(img)