mp-02 commited on
Commit
be6e0a6
·
verified ·
1 Parent(s): ec3d00d

Update cord_inference.py

Browse files
Files changed (1) hide show
  1. cord_inference.py +1 -1
cord_inference.py CHANGED
@@ -4,7 +4,7 @@ from transformers import LayoutLMv3TokenizerFast, LayoutLMv3Processor, LayoutLMv
4
  from PIL import Image, ImageDraw, ImageFont
5
  from utils import OCR, unnormalize_box
6
 
7
- labels = [
8
  "O",
9
  "B-MENU.CNT",
10
  "B-MENU.DISCOUNTPRICE",
 
4
  from PIL import Image, ImageDraw, ImageFont
5
  from utils import OCR, unnormalize_box
6
 
7
+ labels_list = [
8
  "O",
9
  "B-MENU.CNT",
10
  "B-MENU.DISCOUNTPRICE",