Update app.py

app.py

@@ -3,11 +3,12 @@ from PIL import Image
 import tensorflow as tf
 import numpy as np
 import os
+from keras.models import load_model
 
 # Caching the model loading function to optimize performance
 @st.cache_resource
 def load_model():
-    model_path = "captcha.keras"  # Update with the actual model path
+    model_path = ('captcha.keras', custom_objects={'CustomLayer': CustomLayer})  # Update with the actual model path
     return tf.keras.models.load_model(model_path)
 
 # Load the model