Update app.py

app.py

@@ -63,15 +63,20 @@ def prepare_image(img):
     return pred_texts, preds
 
 def decode_batch_predictions(pred):
+    # Misalnya, Anda memiliki daftar karakter yang digunakan dalam model OCR
     characters = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789"  # Update sesuai dengan karakter yang digunakan
     pred_texts = []
     
     for i in range(len(pred)):  # Looping melalui prediksi batch
-        pred_text = ''.join([characters[int(c)] for c in pred[i].tolist() if c != -1])  # Convert pred[i] ke list dan menghindari -1
+        # Pastikan pred[i] adalah array 1D yang berisi nilai-nilai numerik
+        # Gunakan flatten jika pred[i] adalah array 2D atau lebih
+        pred_flat = pred[i].flatten() if isinstance(pred[i], np.ndarray) else pred[i]
+
+        pred_text = ''.join([characters[int(c)] for c in pred_flat if c != -1])  # Menghindari nilai -1
         pred_texts.append(pred_text)
     
     return pred_texts
-
+    
 def run():
     st.title("OCR Model Deployment")