Update app.py

app.py

@@ -5,6 +5,13 @@ import gradio as gr
 import numpy as np
 import tensorflow as tf
 
+# Ensure the model file exists before loading
+model_path = "sleep_cognition_model.h5"
+
+if not os.path.exists(model_path):
+    raise FileNotFoundError(f"⚠️ Model file not found: {model_path}. Please upload it.")
+
+
 # Load the trained model
 model = tf.keras.models.load_model("sleep_cognition_model.h5")