Hugging Face's logo

Spaces:
mp-02
/
LayoutLMv3_for_recepits2
Sleeping

App Files Community
mp-02 commited on
Commit
e1b2c07
·
verified ·
1 Parent(s): d023795

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -16,7 +16,7 @@ model.to(device)
16
 
17
 
18
  def prediction(image):
19
- image = Image.open(image)
20
  boxes, words = OCR(image)
21
  encoding = processor(image, words, boxes=boxes, return_offsets_mapping=True, return_tensors="pt", truncation=True)
22
  offset_mapping = encoding.pop('offset_mapping')
 
16
 
17
 
18
  def prediction(image):
19
+
20
  boxes, words = OCR(image)
21
  encoding = processor(image, words, boxes=boxes, return_offsets_mapping=True, return_tensors="pt", truncation=True)
22
  offset_mapping = encoding.pop('offset_mapping')