Update app.py
Browse files
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
|
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")
|