Update app.py

app.py

@@ -3,14 +3,19 @@ from PIL import Image
 import tensorflow as tf
 import numpy as np
 import os
+from tensorflow.keras.layers import LSTM
+from keras.saving import register_keras_serializable
+
+# Kelas LSTM Kustom
+@register_keras_serializable()
+class CustomLSTM(LSTM):
+    pass
 
 # 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
-
-    # Load the model without custom objects
-    return tf.keras.models.load_model(model_path)
+    return tf.keras.models.load_model(model_path, custom_objects={'CustomLSTM': CustomLSTM})
 
 # Load the model
 model = load_captcha_model()