Update app.py

app.py

@@ -73,10 +73,17 @@ def prepare_image(img):
 
 def decode_batch_predictions(pred):
     pred_texts = []
+    
+    # Loop untuk setiap batch (jika ada lebih dari satu batch)
     for i in range(pred.shape[0]):
-        # Ensure you're iterating over each element in the row pred[i]
-        pred_text = ''.join([characters[int(c)] for c in pred[i] if c not in [-1, 0]])  # Use simple comparison
-        pred_texts.append(pred_text)
+        # Ambil argmax untuk mendapatkan indeks karakter yang diprediksi (yaitu karakter yang paling mungkin)
+        pred_indices = np.argmax(pred[i], axis=-1)  # Cari indeks dengan probabilitas tertinggi
+        
+        # Sekarang kita memetakan indeks ini ke karakter, mengecualikan nilai yang tidak valid seperti -1 atau 0 (padding)
+        pred_text = ''.join([characters[int(c)] for c in pred_indices if c != -1 and c != 0])
+        
+        pred_texts.append(pred_text)  # Simpan teks hasil decoding untuk batch yang sedang diproses
+    
     return pred_texts