Update cord_inference.py

cord_inference.py

@@ -94,11 +94,9 @@ def prediction(image):
     true_words = true_words[1:-1]
     true_confidence_scores = true_confidence_scores[1:-1]
 
-    """
     for i, conf in enumerate(true_confidence_scores):
-        if conf < 0.5 :
-            true_predictions[i] = "O"
-    """            
+        if conf < 0.6 :
+            true_predictions[i] = "O"          
     
     d = {}
     for id, i in enumerate(true_predictions):
@@ -107,6 +105,7 @@ def prediction(image):
         else:
             d[i] = d[i] + ", " + true_words[id]
     d = {k: v.strip() for (k, v) in d.items()}
+    d.pop("O")
 
     # TODO:process the json