WebashalarForML commited on
Commit
6c1cb89
·
verified ·
1 Parent(s): cb90bb0

Update utility/utils.py

Browse files
Files changed (1) hide show
  1. utility/utils.py +2 -2
utility/utils.py CHANGED
@@ -88,7 +88,8 @@ def process_image(image_path, scale=2):
88
 
89
  def ocr_with_paddle(img):
90
  finaltext = ''
91
- ocr = PaddleOCR(lang='en', use_angle_cls=True)
 
92
  # img_path = 'exp.jpeg'
93
  result = ocr.ocr(img)
94
 
@@ -97,7 +98,6 @@ def ocr_with_paddle(img):
97
  finaltext += ' '+ text
98
  return finaltext
99
 
100
-
101
  def extract_text_from_images(image_paths, RESULT_FOLDER):
102
  all_extracted_texts = {}
103
  all_extracted_imgs={}
 
88
 
89
  def ocr_with_paddle(img):
90
  finaltext = ''
91
+ model_dir = os.getenv('PADDLEOCR_MODEL_DIR', '/tmp/.paddleocr')
92
+ ocr = PaddleOCR(lang='en', use_angle_cls=True, det_model_dir=model_dir)
93
  # img_path = 'exp.jpeg'
94
  result = ocr.ocr(img)
95
 
 
98
  finaltext += ' '+ text
99
  return finaltext
100
 
 
101
  def extract_text_from_images(image_paths, RESULT_FOLDER):
102
  all_extracted_texts = {}
103
  all_extracted_imgs={}