Update utility/utils.py

utility/utils.py

@@ -21,6 +21,7 @@ logging.basicConfig(
 )
 
 # Set the PaddleOCR home directory to a writable location
+import os
 
 os.environ['PADDLEOCR_HOME'] = '/tmp/.paddleocr' 
 
@@ -109,11 +110,19 @@ def ocr_with_paddle(img):
     # 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')
+        use_angle_cls=False
+        # 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
     if isinstance(img, str):