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

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +2 -1
inference.py CHANGED
@@ -21,7 +21,8 @@ def prediction(image):
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
-
 
25
  # Esegui l'inferenza con il modello fine-tuned
26
  with torch.no_grad():
27
  outputs = model(**encoding)
 
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
27
  with torch.no_grad():
28
  outputs = model(**encoding)