Update app.py

app.py

@@ -3,12 +3,20 @@ import numpy as np
 import cv2
 from PIL import Image
 import tensorflow as tf
+import os
 
 # Function to load the model
-@st.cache
+@st.cache_resource
 def load_model():
+    # Ensure this path is correct relative to your deployment setup
+    model_path = 'path_to_your_saved_model.h5'  
+    
+    if not os.path.isfile(model_path):
+        st.error(f"Model file not found: {model_path}")
+        return None
+    
     try:
-        model = tf.keras.models.load_model('path_to_your_saved_model.h5')  # Provide the path to your model
+        model = tf.keras.models.load_model(model_path)
         return model
     except Exception as e:
         st.error(f"Error loading model: {e}")