Update sroie_inference.py

sroie_inference.py

@@ -71,9 +71,9 @@ def prediction(image):
     true_words = true_words[1:-1]
     true_confidence_scores = true_confidence_scores[1:-1]
 
-    for i, j in enumerate(true_confidence_scores):
-        if j < 0.8:  #####################################
-            true_predictions[i] = "O"
+    #for i, j in enumerate(true_confidence_scores):
+    #    if j < 0.8:  #####################################
+    #        true_predictions[i] = "O"
 
     d = {}
     for id, i in enumerate(true_predictions):
@@ -94,11 +94,11 @@ def prediction(image):
     image = (blur(image, blur_boxes))
 
     #draw = ImageDraw.Draw(image, "RGBA")
-    #font = ImageFont.load_default()
+    font = ImageFont.load_default()
 
-    #for prediction, box in zip(preds, bboxes):
-    #    draw.rectangle(box)
-    #    draw.text((box[0]+10, box[1]-10), text=prediction, font=font, fill="black", font_size="8")
+    for prediction, box in zip(preds, bboxes):
+       draw.rectangle(box)
+       draw.text((box[0]+10, box[1]-10), text=prediction, font=font, fill="black", font_size="8")
 
     return d, image