Update app.py

app.py

@@ -62,14 +62,14 @@ def prepare_image(img):
 
     return pred_texts, preds
 
-# Define a simple batch decoder (adjust as needed)
 def decode_batch_predictions(pred):
-    # This function should convert the predictions (logits) to text
-    # Modify this function based on your specific character map
+    characters = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789"  # Update sesuai dengan karakter yang digunakan
     pred_texts = []
-    for i in range(pred.shape[0]):
-        pred_text = ''.join([characters[int(c)] for c in pred[i] if c != -1])  # Map to characters
+    
+    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
         pred_texts.append(pred_text)
+    
     return pred_texts
 
 def run():