Update utility/utils.py

utility/utils.py

@@ -113,11 +113,13 @@ def ocr_with_paddle(img):
     #model_dir = os.getenv('PADDLEOCR_MODEL_DIR', '/tmp/.paddleocr')
     #ocr = PaddleOCR(lang='en', use_angle_cls=True, det_model_dir=model_dir)
     #ocr = PaddleOCR(lang='en', use_angle_cls=True, det_model_dir=os.environ['PADDLEOCR_HOME'])
+    logging.info(f"PADDLEOCR_HOME: {os.environ['PADDLEOCR_HOME']}")
     ocr = PaddleOCR(
-                    lang='en', 
-                    use_angle_cls=True, 
-                    rec_model_dir=os.path.join(os.environ['PADDLEOCR_HOME'], 'whl/rec/en/en_PP-OCRv4_rec_infer'),
-                    det_model_dir=os.path.join(os.environ['PADDLEOCR_HOME'], 'whl/det')
+        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')
     )
 
     result = ocr.ocr(img)