mp-02 commited on
Commit
9b9b439
1 Parent(s): f2fefda

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +3 -3
inference.py CHANGED
@@ -53,9 +53,9 @@ def prediction(image):
53
  true_words = true_words[1:-1]
54
  true_confidence_scores = true_confidence_scores[1:-1]
55
 
56
- for i, j in enumerate(true_confidence_scores):
57
- if j < 0.5:
58
- true_predictions[i] = "O"
59
 
60
  d = {}
61
  for id, i in enumerate(true_predictions):
 
53
  true_words = true_words[1:-1]
54
  true_confidence_scores = true_confidence_scores[1:-1]
55
 
56
+ #for i, j in enumerate(true_confidence_scores):
57
+ # if j < 0.5:
58
+ # true_predictions[i] = "O"
59
 
60
  d = {}
61
  for id, i in enumerate(true_predictions):