mp-02 commited on
Commit
0453cbe
·
verified ·
1 Parent(s): 3f72c9d

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -20,7 +20,7 @@ def prediction(image):
20
 
21
  boxes, words = OCR(image)
22
  # Preprocessa l'immagine e il testo con il processore di LayoutLMv3
23
- encoding = processor(image, words, boxes=boxes, return_tensors="pt", padding="max_length", truncation=True)
24
  offset_mapping = encoding.pop('offset_mapping')
25
 
26
  # Esegui l'inferenza con il modello fine-tuned
 
20
 
21
  boxes, words = OCR(image)
22
  # Preprocessa l'immagine e il testo con il processore di LayoutLMv3
23
+ encoding = processor(image, words, boxes=boxes, return_offsets_mapping=True, return_tensors="pt", truncation=True)
24
  offset_mapping = encoding.pop('offset_mapping')
25
 
26
  # Esegui l'inferenza con il modello fine-tuned