Update app.py

app.py

@@ -1,14 +1,18 @@
 import streamlit as st
 from PIL import Image
 import tensorflow as tf
+from tensorflow.keras.models import load_model
 import numpy as np
 import os
+from tensorflow.keras.layers import LSTM
 
 # Caching the model loading function to optimize performance
 @st.cache_resource
 def load_captcha_model():
     model_path = "captcha_ocr_model.h5"  # Update with the actual CAPTCHA model path
-    return tf.keras.models.load_model(model_path)
+
+    # Load the model with custom objects
+    return tf.keras.models.load_model(model_path, custom_objects={'LSTM': LSTM})
 
 # Load the model
 model = load_captcha_model()
@@ -26,7 +30,6 @@ def prepare_captcha_image(img):
         predictions = model.predict(img_array)
 
         # Decode predictions assuming the model outputs probabilities
-        # Modify this part based on your specific model's output
         decoded_captcha = ''.join([chr(np.argmax(pred) + ord('A')) for pred in predictions])
 
         return decoded_captcha, predictions