Fixed OCR decoding error - ensured indices are iterable

app.py

@@ -24,15 +24,23 @@ def predict_text(image):
     print(prediction)  # Check what the model is outputting
     predicted_text = decode_prediction(prediction)  # Implement your decoding logic
     return predicted_text
+
 import string
+import numpy as np
 
 # Define possible characters in prescriptions (letters, numbers, etc.)
 CHARACTER_SET = string.ascii_letters + string.digits + " .,-/()"
 
-# Placeholder function for decoding (depends on model architecture)
 def decode_prediction(prediction):
-    indices = np.argmax(prediction, axis=-1)  # Get index of highest probability for each character
-    text = "".join([CHARACTER_SET[i] for i in indices[0]])  # Convert indices to characters
+    indices = np.argmax(prediction, axis=-1)  # Get indices of highest probabilities
+
+    # Ensure indices is iterable (check if it's a single number or array)
+    if indices.ndim == 1:  # If it's a 1D array
+        indices = indices.tolist()  # Convert to a list
+    elif indices.ndim == 0:  # If it's a single number
+        indices = [indices.item()]  # Wrap it in a list
+
+    text = "".join([CHARACTER_SET[i] for i in indices])  # Convert indices to characters
     return text
 
 # Gradio UI