Update app.py

app.py

@@ -3,10 +3,18 @@ import numpy as np
 import cv2
 from tensorflow.keras.models import load_model
 
-# Load the trained model
+
+from tensorflow.keras.layers import InputLayer
+
+# Custom object scope to handle InputLayer configuration
+custom_objects = {'InputLayer': InputLayer}
+
 @st.cache_resource
 def load_ocular_model():
-    return load_model('odir_cnn_model.h5')
+    return load_model('odir_cnn_model.h5', custom_objects=custom_objects)
+
+model = load_ocular_model()
+
 
 model = load_ocular_model()