Spaces:
Sleeping
Sleeping
Update inference.py
Browse files- 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)
|