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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -75,9 +75,10 @@ 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
 
82
  def run():
83
  st.title("OCR Model Deployment")
 
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 c not in [-1, 0]]) # Use simple comparison
79
  pred_texts.append(pred_text)
80
  return pred_texts
81
+
82
 
83
  def run():
84
  st.title("OCR Model Deployment")