Update inference.py

inference.py

@@ -53,10 +53,6 @@ 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
@@ -66,7 +62,8 @@ def prediction(image):
             d[i] = true_words[id]
         else:
             d[i] = d[i] + ", " + true_words[id]
-    d = {k: v.strip() for (k, v) in d.items()}
+    
+    #d = {k: v.strip() for (k, v) in d.items()}
 
     if "O" in d: d.pop("O")