mp-02 commited on
Commit
96db0e7
·
verified ·
1 Parent(s): 808fea7

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -63,7 +63,7 @@ def prediction(image):
63
  else:
64
  d[i] = d[i] + ", " + true_words[id]
65
 
66
- #d = {k: v.strip() for (k, v) in d.items()}
67
 
68
  if "O" in d: d.pop("O")
69
 
 
63
  else:
64
  d[i] = d[i] + ", " + true_words[id]
65
 
66
+ d = {k: v.strip() for (k, v) in d.items()}
67
 
68
  if "O" in d: d.pop("O")
69