WebashalarForML commited on
Commit
b1e3b03
1 Parent(s): 9994261

Update utility/utils.py

Browse files
Files changed (1) hide show
  1. utility/utils.py +2 -1
utility/utils.py CHANGED
@@ -162,7 +162,8 @@ def extract_text_from_images(image_paths):
162
 
163
  # Save the image with boxes
164
  result_image_path = os.path.join(RESULT_FOLDER, f'result_{os.path.basename(image_path)}')
165
- img_with_boxes.save(result_image_path)
 
166
 
167
  # Store the text and image result paths
168
  all_extracted_texts[image_path] = result
 
162
 
163
  # Save the image with boxes
164
  result_image_path = os.path.join(RESULT_FOLDER, f'result_{os.path.basename(image_path)}')
165
+ #img_with_boxes.save(result_image_path)
166
+ cv2.imwrite(result_image_path, img_with_boxes)
167
 
168
  # Store the text and image result paths
169
  all_extracted_texts[image_path] = result