Update utility/utils.py

utility/utils.py

@@ -23,7 +23,7 @@ logging.basicConfig(
 # Set the PaddleOCR home directory to a writable location
 import os
 
-os.environ["PADDLEOCR_HOME"] = "/app/paddleocr_models"
+os.environ['PADDLEOCR_HOME'] = '/tmp/.paddleocr' 
 
 RESULT_FOLDER = 'static/results/'
 JSON_FOLDER = 'static/json/'
@@ -108,20 +108,20 @@ 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,
-        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'
+        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(
+    #     use_angle_cls=True,
+    #     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