Update inference.py

inference.py

@@ -36,7 +36,7 @@ def create_json_output(words, labels, boxes):
     json_output = json.dumps(output, indent=4)
     return json_output
 
-def prediction()
+def prediction(Image image)
 
     boxes, words = OCR(image)
     # Preprocessa l'immagine e il testo con il processore di LayoutLMv3
@@ -56,6 +56,9 @@ def prediction()
     confidence_scores = probabilities.max(-1).values.squeeze().tolist()
     # Crea il JSON usando i risultati ottenuti
     json_result = create_json_output(words, labels, boxes)
+
+    draw = ImageDraw.Draw(image, "RGBA")
+    font = ImageFont.load_default()
     
     for prediction, box, confidence in zip(true_predictions, true_boxes, true_confidence_scores):
             draw.rectangle(box)