Update inference.py

inference.py

@@ -69,7 +69,7 @@ def prediction(image):
     draw = ImageDraw.Draw(image, "RGBA")
     font = ImageFont.load_default()
     
-    for prediction, box, confidence in zip(true_predictions, token_boxes, confidence_scores):
+    for prediction, box, confidence in zip(true_predictions, true_boxes, true_confidence_scores):
             draw.rectangle(box)
             draw.text((box[0]+10, box[1]-10), text=str(prediction)+ ", "+ str(confidence), font=font, fill="black", font_size="15")