kaydee commited on
Commit
1996666
1 Parent(s): e18f77b

Update extract_info.py

Browse files
Files changed (1) hide show
  1. extract_info.py +2 -1
extract_info.py CHANGED
@@ -15,7 +15,8 @@ from transformers import AutoModelForTokenClassification
15
  from datasets import load_dataset
16
  from PIL import Image, ImageDraw, ImageFont
17
  import pytesseract
18
- #from imgprocessing import processed_image
 
19
 
20
  processor = AutoProcessor.from_pretrained("kaydee/layoutlmv3-wildreceipt", apply_ocr=True)
21
  model = AutoModelForTokenClassification.from_pretrained("kaydee/layoutlmv3-wildreceipt")
 
15
  from datasets import load_dataset
16
  from PIL import Image, ImageDraw, ImageFont
17
  import pytesseract
18
+
19
+ pytesseract.pytesseract.tesseract_cmd = r'C:\\Program Files\\Tesseract-OCR\\tesseract.exe'
20
 
21
  processor = AutoProcessor.from_pretrained("kaydee/layoutlmv3-wildreceipt", apply_ocr=True)
22
  model = AutoModelForTokenClassification.from_pretrained("kaydee/layoutlmv3-wildreceipt")