Update extract_info.py
Browse files- 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 |
-
|
|
|
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")
|