mp-02 commited on
Commit
ff7a14a
1 Parent(s): 173c871

Update sroie_inference.py

Browse files
Files changed (1) hide show
  1. sroie_inference.py +1 -1
sroie_inference.py CHANGED
@@ -78,7 +78,7 @@ def prediction(image):
78
  d = {}
79
  for id, i in enumerate(true_predictions):
80
  if i not in d.keys():
81
- d[i] = true_words[d]
82
  else:
83
  d[i] = d[i] + ", " + true_words[id]
84
  d = {k: v.strip() for (k, v) in d.items()}
 
78
  d = {}
79
  for id, i in enumerate(true_predictions):
80
  if i not in d.keys():
81
+ d[i] = true_words[id]
82
  else:
83
  d[i] = d[i] + ", " + true_words[id]
84
  d = {k: v.strip() for (k, v) in d.items()}