Update inference.py

inference.py

@@ -53,6 +53,10 @@ 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.5:  
+            true_predictions[i] = "O"
+
     d = {}
     for id, i in enumerate(true_predictions):
         #rimuovo i prefissi