Update app.py

app.py

@@ -5,9 +5,8 @@ from PIL import Image
 import sqlite3
 import cv2
 import numpy as np
-from tensorflow.keras.models import load_model
-from datetime import datetime
-
+from tensorflow.keras.models import load_model  # Importing load_model
+from datetime import datetime  # Importing datetime
 
 # Constants
 HOME_DIR = os.getcwd()  # Home directory to save images (root directory)
@@ -37,7 +36,14 @@ except Exception as e:
 
 # Load the emotion detection model
 try:
+    # Check if model file exists
+    if not os.path.exists(EMOTION_MODEL_FILE):
+        st.error(f"Error: Emotion model file '{EMOTION_MODEL_FILE}' not found!")
+        st.stop()
+
+    # Load the model
     emotion_model = load_model(EMOTION_MODEL_FILE)  # Load the emotion model
+    st.success("Emotion detection model loaded successfully!")
 except Exception as e:
     st.error(f"Error loading emotion model: {e}")
     st.stop()