Update cord_inference.py

cord_inference.py

@@ -59,13 +59,6 @@ def prediction(image):
     l_words = []
     bboxes = []
 
-    for i, j in enumerate(true_predictions):
-        if true_confidence_scores[i] < 0.8:  #####################################àà
-            true_predictions[i] = "O"
-        preds.append(true_predictions[i])
-        l_words.append(true_words[i])
-        bboxes.append(true_boxes[i])
-
     d = {}
     for id, i in enumerate(preds):
         if i not in d.keys():
@@ -79,8 +72,8 @@ def prediction(image):
     draw = ImageDraw.Draw(image, "RGBA")
     font = ImageFont.load_default()
 
-    for prediction, box in zip(preds, bboxes):
+    for prediction, box, confidence in zip(preds, bboxes, true_confidence_scores):
         draw.rectangle(box)
-        draw.text((box[0]+10, box[1]-10), text=prediction, font=font, fill="black")
+        draw.text((box[0]+10, box[1]-10), text=prediction+ ", "+ confidence, font=font, fill="black", font_size="15")
 
     return d, image