mp-02 commited on
Commit
0581571
·
verified ·
1 Parent(s): 4f4c048

Update cord_inference.py

Browse files
Files changed (1) hide show
  1. cord_inference.py +4 -2
cord_inference.py CHANGED
@@ -55,10 +55,12 @@ def prediction(image):
55
  true_words = true_words[1:-1]
56
  true_confidence_scores = true_confidence_scores[1:-1]
57
 
 
58
  for i, conf in enumerate(true_confidence_scores):
59
- if conf < 0.6 :
60
  true_predictions[i] = "O"
61
-
 
62
  d = {}
63
  for id, i in enumerate(true_predictions):
64
  if i not in d.keys():
 
55
  true_words = true_words[1:-1]
56
  true_confidence_scores = true_confidence_scores[1:-1]
57
 
58
+ """
59
  for i, conf in enumerate(true_confidence_scores):
60
+ if conf < 0.5 :
61
  true_predictions[i] = "O"
62
+ """
63
+
64
  d = {}
65
  for id, i in enumerate(true_predictions):
66
  if i not in d.keys():