Reaumur commited on
Commit
b00e317
·
verified ·
1 Parent(s): 4d0b1e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -74,7 +74,8 @@ def prepare_image(img):
74
  def decode_batch_predictions(pred):
75
  pred_texts = []
76
  for i in range(pred.shape[0]):
77
- pred_text = ''.join([characters[int(c)] for c in pred[i] if int(c) != -1 and int(c) != 0]) # Convert `c` to int for comparison
 
78
  pred_texts.append(pred_text)
79
  return pred_texts
80
 
 
74
  def decode_batch_predictions(pred):
75
  pred_texts = []
76
  for i in range(pred.shape[0]):
77
+ # Ensure you're iterating over each element in the row pred[i]
78
+ pred_text = ''.join([characters[int(c)] for c in pred[i] if not np.any(c == [-1, 0])]) # Use np.any to check for -1 and 0
79
  pred_texts.append(pred_text)
80
  return pred_texts
81