kaydee commited on
Commit
9b5c822
1 Parent(s): 95f34c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
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")