Spaces:
Sleeping
Sleeping
Update cord_inference.py
Browse files- cord_inference.py +2 -2
cord_inference.py
CHANGED
@@ -55,11 +55,11 @@ def prediction(image):
|
|
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):
|
|
|
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):
|