Update app.py
Browse files
app.py
CHANGED
@@ -17,12 +17,10 @@ from transformers import AutoModelForTokenClassification
|
|
17 |
from datasets import load_dataset
|
18 |
from PIL import Image, ImageDraw, ImageFont
|
19 |
|
20 |
-
|
21 |
processor = AutoProcessor.from_pretrained("kaydee/layoutlmv3-wildreceipt", apply_ocr=True)
|
22 |
model = AutoModelForTokenClassification.from_pretrained("kaydee/layoutlmv3-wildreceipt")
|
23 |
|
24 |
-
|
25 |
-
|
26 |
# load image example
|
27 |
dataset = load_dataset("kaydee/wildreceipt", split="test")
|
28 |
dataset[20]["image"].convert("RGB").save("example1.png")
|
|
|
17 |
from datasets import load_dataset
|
18 |
from PIL import Image, ImageDraw, ImageFont
|
19 |
|
20 |
+
pytesseract.pytesseract.tesseract_cmd = r'C:\\Program Files\\Tesseract-OCR\\tesseract.exe'
|
21 |
processor = AutoProcessor.from_pretrained("kaydee/layoutlmv3-wildreceipt", apply_ocr=True)
|
22 |
model = AutoModelForTokenClassification.from_pretrained("kaydee/layoutlmv3-wildreceipt")
|
23 |
|
|
|
|
|
24 |
# load image example
|
25 |
dataset = load_dataset("kaydee/wildreceipt", split="test")
|
26 |
dataset[20]["image"].convert("RGB").save("example1.png")
|