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