Update cord_inference.py

cord_inference.py

@@ -54,11 +54,13 @@ def prediction(image):
     true_confidence_scores = true_confidence_scores[1:-1]
 
     for i, conf in enumerate(true_confidence_scores):
-        if conf < 0.6 :
+        if conf < 0.5 :
             true_predictions[i] = "O"          
     
     d = {}
     for id, i in enumerate(true_predictions):
+        if i != "O":
+            i = i[2:]
         if i not in d.keys():
             d[i] = true_words[id]
         else: