Spaces:

kartai
/

CAD-AID

Runtime error

juliajo commited on May 14, 2025

Commit

56774cf

verified ·

1 Parent(s): 61d87f0

Update src/utils/text_ocr.py

Files changed (1) hide show

src/utils/text_ocr.py CHANGED Viewed

@@ -110,7 +110,7 @@ def pytesseract_ocr_detection(image_path):
     gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
     _, thresh = cv2.threshold(gray, 180, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)
-    data = pytesseract.image_to_data(thresh, lang='nor', output_type=pytesseract.Output.DICT)
     detected_text = []
     n_boxes = len(data['text'])

     gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
     _, thresh = cv2.threshold(gray, 180, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)
+    data = pytesseract.image_to_data(thresh, output_type=pytesseract.Output.DICT)
     detected_text = []
     n_boxes = len(data['text'])